Senior Hardware-Aware ML Scientist for Large-Scale Training

Annapurna Labs (U.S.) Inc.; Cupertino (CA); On-site; USD 180,000 - 250,000; Full time

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, California seeks an Applied Scientist to push hardware-aware ML for Trainium and Inferentia.

You will own end-to-end model and algorithm decisions from research to production, publishing work at top venues and collaborating with distinguished engineers and scientists in a strategic AWS growth area. You will work on low-precision training/inference, Trn-friendly architectures, system-aware optimizers, and GenAI for systems, influencing what gets built into

Qualifications

• PhD or MS with 6+ years of applied research experience.
• Experience programming in Java, C++, Python or related language.
• 3+ years building ML models for business applications.
• Experience with neural deep learning methods.

Responsibilities

• Own scientific problems end-to-end from research through production impact.
• Develop production-quality code in PyTorch or JAX for large-scale systems.
• Collaborate with foundation-model, engineering, and hardware teams.
• Mentor scientists and interns and help shape technical roadmap.
• Publish research at top venues and engage with the scientific community.

Skills

Java/C++/Python

Deep learning

ML models

Education

PhD or MS + 6+ years research

Tools

R

scikit-learn

Spark MLLib

MxNet

TensorFlow

NumPy

SciPy

Hadoop

Spark

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, California seeks an Applied Scientist to push hardware-aware ML for Trainium and Inferentia.

You will own end-to-end model and algorithm decisions from research to production, publishing work at top venues and collaborating with distinguished engineers and scientists in a strategic AWS growth area. You will work on low-precision training/inference, Trn-friendly architectures, system-aware optimizers, and GenAI for systems, influencing what gets built into