Software Development Engineer, AWS Neuron
Core
Design, build, test, deploy, and maintain the AWS Neuron compiler to optimize deep-learning models on AWS Inferentia and Trainium chips.
Role type
Senior IC compiler engineer (ML accelerators)
Builds
Next-generation compiler for PyTorch, TensorFlow, JAX, and large language models
Domain
Cloud computing + ML compiler infrastructure
Deliverable
production ML models
Required skills
C++, Java, compiler design, system architecture, scaling, reliability
Preferred skills
LLVM, MLIR, OpenXLA, StableHLO, Bazel, CMake, PyTorch, TensorFlow, JAX, TVM
Technologies
LLVM, MLIR, OpenXLA, StableHLO, Bazel, CMake, PyTorch, TensorFlow, JAX, TVM
Responsibilities
Develop compiler optimization and verification passes; build tools for analyzing numerical errors; resolve compiler defects; collaborate with chip architects and ML teams; contribute to open-source projects; participate in pre-silicon design
Seniority
Senior, hands-on IC
