Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Core
Building next-generation compilers to optimize ML models (PyTorch, TensorFlow, JAX) for deployment on AWS Inferentia and Trainium chips.
Role type
Senior IC compiler engineer (ML accelerators)
Builds
High-performance Neuron compiler for AWS cloud ML workloads
Domain
Cloud computing + ML compiler optimization
Deliverable
production ML models
Required skills
C++/Java, compiler design, ML framework internals, system architecture
Preferred skills
OpenXLA, StableHLO, MLIR, PyTorch, JAX, Bazel/CMake, GPU/accelerator programming
Technologies
AWS Inferentia, AWS Trainium, PyTorch, TensorFlow, JAX, OpenXLA, StableHLO, MLIR, LLVM, Bazel, CMake
Responsibilities
Design and implement compiler optimization and verification passes, build features to surface accelerator peculiarities to developers, implement tools to analyze numerical errors, resolve root causes of compiler defects, participate in design discussions and code reviews
Seniority
Senior, hands-on IC
