Applied Scientist, AWS Neuron Science - Core Algorithm
Amazon· AWS Neuron Science - Core Algorithm
Cupertino$171.6k - $222.2kFull-timePosted yesterday
About the role
Applied Scientist role focusing on hardware-aware machine learning for AWS Trainium and Inferentia. Ownership of end-to-end model and algorithm decisions, collaboration with foundation-model, engineering, and hardware teams, with opportunities to publish research and mentor others.
Responsibilities
- Own scientific problems end-to-end - from research and experimentation through production impact - applying rigorous evaluation to complex, ill-defined problems at large scale.
- Develop production-quality code in PyTorch or JAX and integrate scientific components into large-scale training and inference systems with operational excellence and efficient resource usage.
- Partner with foundation-model, engineering, and hardware-architecture teams so findings inform what gets built into Trainium and shipped in the product stack.
- Mentor fellow scientists and interns, give constructive peer reviews, and help shape team goals, priorities, and the technical roadmap.
- Author and publish research at top peer-reviewed venues (ICLR, NeurIPS, ICML, MLSys) and engage the broader scientific community.
Qualifications
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- 3+ years of building models for business application experience
- Experience programming in Java, C++, Python or related language
- Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning
Skills
Required:PyTorch or JAXFoundation-model collaboration
Nice to have:Mentoring
Benefits
- Health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans)
- EAP, Mental Health Support, Medical Advice Line
- Flexible Spending Accounts
- Adoption and Surrogacy Reimbursement coverage
- 401(k) matching
- Paid time off
- Parental leave
Last updated Oct 25, 2026. Interested? Apply on the company site.
More open roles at Amazon
- IT Support Associate II, Ops Tech Solutions (OTS)yesterdayPlainfield, Indiana · Operations IT Support Engineering
- Head of Energy and Water, AWS LegalyesterdayHybrid (Seattle, WA) · Legal
- Software Development Engineer, Creative Asset Library, CreativeXyesterdaySeattle · Software Development
- Software Development Engineer, Creative Asset Library, CreativeXyesterdayHybrid (Seattle, WA) · Software Development
- IT Support Associate II, OTSyesterdayIrving, TX · Operations IT Support Engineering