Applied Scientist, Neuron ARG, Annapurna ML

Amazon

Seattle, WA, US
171,600.00 - 222,200.00 usd annually py
On-site
Machine learning compiler technology
Program analysis and synthesis engines
Deep learning systems software
This role focuses on raising the code quality bar in Amazon's state-of-the-art deep learning compiler stack by working at the intersection of AI and program analysis

Job Summary

  • This role focuses on raising the code quality bar in Amazon's state-of-the-art deep learning compiler stack by working at the intersection of AI and program analysis.
  • The successful candidate will design, develop, and deploy analyzers for ML compiler stages while architecting business-critical tooling for the Neuron team.
  • You will collaborate closely with a talented engineering team to innovate on behalf of customers using Trainium, the forefront of advanced ML capabilities.

Matching Summary

This role focuses on raising the code quality bar in Amazon's state-of-the-art deep learning compiler stack by working at the intersection of AI and program analysis.

Salary

171,600.00 - 222,200.00 USD annually

Skills & Requirements

Must-have

  • Machine Learning compiler technology
  • Program analysis and synthesis engines
  • Deep learning systems software
  • Custom ML accelerator optimization

Nice-to-have

  • Experience with PyTorch and JAX frameworks
  • Background in theorem provers
  • Mentoring experienced scientists and engineers
  • Research publication experience

Key Requirements

  • Strong programming language expertise
  • Compiler and program analyzer experience
  • Technical credibility as an AWS engineer

Work Rights

Not specified

Tailored Resume

Cover Letter