Drimva

Amazon

Applied Scientist, Neuron ARG, Annapurna ML

$143K–$222K

Seattle, WA +1 · Full-time · Posted Sep 5, 2026

Location
Seattle, WA
Cupertino, CA
Experience
4+ years with a Master's degree; alternatively, a PhD is listed.
Education
PhD, or Master's degree and 4+ years of CS, CE, ML, or related field experience.

About the job

Develop and deploy analyzers and tooling for deep-learning compiler stages and intermediate representations, advancing program analysis and compiler quality for ML workloads on AWS Neuron.

What you'll do

  • Design, develop, and deploy analyzers for ML compiler stages and compiler IRs.
  • Architect and implement business-critical tooling; publish research.
  • Collaborate with scientists and engineers and mentor team members.

What they ask for

  • PhD, or Master's degree and 4+ years of CS, CE, ML, or related field experience.
  • Experience with patents or publications at top-tier peer-reviewed conferences or journals.
  • Programming experience in Java, C++, Python, or a related language.
  • Experience in at least one of: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, or high-performance computing.

Nice to have

  • PhD in computer science, computer engineering, or a related field.
  • Background in machine learning and AI accelerators.

Pay

171,600.00 - 222,200.00 USD annually (base salary); 142,800.00 - 193,200.00 USD annually (base salary)

Tools

Java, C++, Python, PyTorch, JAX, Trainium

AI focus

Applies artificial intelligence and program analysis to deep-learning compilers and ML workloads; works with AI accelerators.

Model-written summary · the original posting is authoritative

First seen Oct 20, 2026 · Last verified Nov 5, 2026 (UTC)