Senior ML Compiler Engineer, Neuron
Core
Design and build compiler features for AWS Neuron compilers to handle large-scale ML workloads on Trainium hardware, including frontend, IR, optimization, and code generation.
Role type
Senior IC ML Compiler Engineer
Builds
Neuron compiler (NKI) and runtime for AWS Trainium instances
Domain
Cloud computing / Machine Learning hardware acceleration
Deliverable
production ML models
Required skills
Compiler design, hardware architecture, ML frameworks (PyTorch, JAX), system design, code review, mentoring
Preferred skills
Full SDLC, pre-silicon design, research publication
Technologies
Neuron SDK, NKI, Trainium, PyTorch, JAX
Responsibilities
Design and build compiler features spanning frontend to code generation; own customer-critical features end-to-end; partner with ML teams to translate workloads into compiler requirements; contribute to pre-silicon hardware design; review code and mentor engineers.
Seniority
Senior, hands-on IC with leadership responsibilities