Machine Learning Researcher - Springtail

Astera Institute; California; Híbrido; USD 150.000 - 300.000; Tempo integral

Resumo da vaga

A pioneering research organization is seeking a Machine Learning Researcher to enhance data-efficient model induction methods. This role involves hypothesizing improvements for generalization performance and contributing to a robust code base. Candidates should hold a relevant Master's degree and demonstrate proficiency in Pytorch alongside experience in collaborative research. The position is hybrid, based in California with a salary ranging from $150K to $300K depending on experience.

Qualificações

• Masters or equivalent in machine learning, mathematics, or equivalent fields.
• Fluency with Pytorch, experience with JAX, CUDA, or Triton.
• Demonstrated ability to do fundamental research.

Responsabilidades

• Hypothesize, test, and refine methods to improve generalization performance.
• Analyze runtime inference in gradient-trained networks.
• Contribute to a performant and documented code base.

Conhecimentos

Pytorch

Teamwork

Mathematical thinking

Empirical thinking

Formação

Masters or equivalent in machine learning or mathematics

Ferramentas

JAX

CUDA

Triton

Descrição da vaga

Astera Institute

Machine Learning Researcher - Springtail

Position Summary

Datasets in many areas, science in particular, are often small, heterogeneous, and expensive. Human scientists can take these datasets and generate models to describe them, but this process of model induction is labor-intensive and error-prone. Machine learning is a general and scalable solution, but it is not uniformly sample efficient.

The Astera Institute is seeking a Machine Learning Researcher to help surmount this barrier with new architectures for data-efficient and general model induction. This includes bootstrapped program synthesis, along with components for a system that synthesizes its own learning algorithms - a machine learning strange loop. This is a full time position that reports to Timothy Hanson.

Responsibilities:

• Hypothesize, test, and refine means of improving generalization performance of common architectural elements, including different forms of attention. This includes devising controlled datasets to elucidate e.g. learning order & learned representations.
• Think both mathematically and empirically about problems of runtime inference in gradient-trained networks, with an eye to the extensive literature on statistical learning and an open mind to the many forms of constrained optimization.
• Contribute to a well-documented and well-instrumented code base that is performant where necessary yet expeditious where experimental throughput demands.

Qualifications And Experience

• Masters or equivalent in machine learning, mathematics, or equivalent fields (strong candidates from neuroscience are encouraged to apply).
• Fluency with Pytorch, and familiarity with JAX, CUDA, and/or Triton + their open-source ecosystems.
• Demonstrated ability do fundamental research.
• Demonstrated ability to work in teams.

About Astera

Astera Institute is a new kind of research organization that operates outside the constraints of markets or academia, and is itself an experiment in how research organizations can move faster, take smarter risks, and create lasting public goods. Our key programs include:

• Our Residency is a one to two-year program that supports extraordinary scientists who are producing public goods. We support not only residents’ salaries, but also budgets for them to hire a team and meet other needs to accelerate their work.
• Radial supports biotechnology-focused projects and is redefining how science should be done.
• Obelisk does fundamental research in Artificial General Intelligence to ensure it brings good to humanity.

Astera is uniquely positioned to shape the course of rapid scientific and technological innovation because of:

• A $2.5B endowment, which we intend to deploy boldly rather than preserve in perpetuity.
• A small, agile team: We are staffed much more leanly than other foundations, and aim to keep bureaucracy as minimal as possible. It's more accurate to think of us as a startup than a traditional foundation.
• High risk tolerance: We take an experimental approach to our programming that means we expect projects to fail with some frequency.
• Disinterest in consensus: we add value by supporting projects, programs, and people that other organizations can’t or won’t.

Location

This position is hybrid at our office in Emeryville, CA. Some travel may be required from time‑to‑time for in‑person collaboration and work.

Compensation

The posted salary range is based on location in the Bay Area. The successful candidate will receive a competitive compensation package, commensurate with their experience and location.

Compensation Range: $150K - $300K