Astera Institute

Machine Learning Researcher

Astera Institute

full-time

Posted on:

Location Type: Hybrid

Location: Emeryville • California • United States

Visit company website

Explore more

✨ AI Apply
Apply

Salary

💰 $150,000 - $300,000 per year

Tech Stack

PyTorch

About the role

  • Hypothesize, test, and refine means of improving generalization performance of common architectural elements
  • Think both mathematically and empirically about problems of runtime inference in gradient-trained networks
  • Contribute to a well-documented and well-instrumented code base that is performant where necessary yet expeditious where experimental throughput demands

Requirements

  • Masters or equivalent in machine learning, mathematics, or equivalent fields
  • Fluency with Pytorch
  • Familiarity with JAX, CUDA, and/or Triton + their open-source ecosystems
  • Demonstrated ability do fundamental research
  • Demonstrated ability to work in teams
Benefits
  • Competitive compensation package
  • Professional development opportunities
  • Well-documented and well-instrumented code base
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
machine learningmathematicsPytorchJAXCUDATritonruntime inferencegradient-trained networkscode performanceexperimental throughput
Soft Skills
fundamental researchteamwork
Certifications
Masters in machine learningMasters in mathematics