Machine Learning Researcher Springtail

A 501(c)(3) science-and-technology research institute/foundation operating direct Neuro-AI research, life-sciences programs through Radial, and an open-science residency program.

Emeryville, CA, United States
About Astera Institute

Astera Institute supports and operates open public-goods work for science and technology. Its current activities include neuroscience-informed AGI research, AI-enabled life-sciences programs, open datasets and publishing infrastructure, and residencies for early-stage projects.

View jobs by Astera Institute

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will research and develop architectures for data-efficient model induction, including bootstrapped program synthesis. You will hypothesize, test, and refine approaches to improve generalization and attention mechanisms, devise controlled datasets, investigate runtime inference and constrained optimization, and contribute to a documented, instrumented codebase.

Requirements

  • Master's degree or equivalent in machine learning, mathematics, or an equivalent field
  • Fluency with PyTorch
  • Familiarity with JAX, CUDA, Triton, or their open-source ecosystems
  • Demonstrated ability to conduct fundamental research
  • Demonstrated ability to work in teams

Responsibilities

  • Hypothesize, test, and refine approaches to improve generalization performance of architectural elements including attention
  • Devise controlled datasets to study learning order and learned representations
  • Investigate runtime inference in gradient-trained networks using statistical-learning and constrained-optimization approaches
  • Contribute to a well-documented, well-instrumented, performant codebase