Senior Hardware-Aware ML Scientist for Large-Scale Training

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