Machine Learning Researcher - Springtail

Stack

Responsibilities

• ▹Improve generalization performance of architectural elements like attention
• ▹Design controlled datasets to study learning order and representations
• ▹Research runtime inference in gradient-trained networks
• ▹Contribute to a well-documented, performant codebase

Requirements

• ▹Master's degree in ML, mathematics, or related field
• ▹Fluency with PyTorch
• ▹Familiarity with JAX, CUDA, and/or Triton
• ▹Demonstrated ability to do fundamental research
• ▹Demonstrated ability to work in teams

Nice to have

• ▹Background in neuroscience

Soft skills

What we offer

• ▹Competitive compensation package

About the company

Astera Institute is a private foundation accelerating science and technology, operating projects like high-velocity startups.