Drimva

Amazon

ML Compiler Engineer, Annapurna Labs

$144K–$224K

Seattle, WA +1 · Full-time · Posted May 21, 2026

Location
Seattle, WA
Cupertino, CA
Experience
3+ years of non-internship professional software development experience; 2+ years of experience in one or more specified optimization, graph-theory, hardware, or resource-management areas. Preferred: 3+ years of full software development life cycle experience.
Education
Bachelor's degree in Computer Science, Engineering, Mathematics, or a related field; preferred M.S. or Ph.D. in computer science or related field.

About the job

Develop and optimize a deep-learning compiler backend for AWS machine-learning accelerators, improving system performance through compiler design and hardware-software collaboration.

What you'll do

  • Design, implement, test, deploy, and maintain compiler software and backend features.
  • Work on areas such as instruction scheduling, memory allocation, graph partitioning, code generation, and hardware bring-up.
  • Collaborate with Runtime, Frameworks, and Hardware teams to optimize system-wide performance.

What they ask for

  • Bachelor's degree in Computer Science, Engineering, Mathematics, or a related field.
  • 3+ years of non-internship professional software development experience.
  • Proficiency with 1 or more programming languages (C++ preferred).
  • 2+ years of experience in optimization algorithms, graph-theory, hardware bring-up, FPGA placement and routing algorithms, or hardware resource management.

Nice to have

  • M.S. or Ph.D. in computer science or related field.
  • Experience with LLVM and/or MLIR.

Pay

165,200.00 - 223,600.00 USD annually (base salary); 143,700.00 - 194,400.00 USD annually (base salary)

Tools

C++, LLVM, MLIR, TensorFlow, PyTorch, JAX, Inferentia, Trainium

AI focus

Develop compiler technology that optimizes deep-learning models, including Large Language and Vision models, for AWS machine-learning accelerators.

Model-written summary · the original posting is authoritative

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