Amazon ML Compiler Engineer Annapurna Labs 

United States, California, Cupertino 

441951649

You: As a Machine Learning Compiler Engineer I on the AWS Neuron Compiler team, you will be supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. Architecting and implementing business-critical features, publish cutting-edge research, and contributing to a brilliant team of experienced engineers excites and challenges you. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects.A background in compiler development is strongly preferred. A background in Machine Learning and AI accelerators is preferred, but not required.Key job responsibilitiesWriting requirements capturing documents, design documents, integration test plans, and deployment plans

- To qualify, applicants should have earned (or will earn) a Bachelors or Masters degree between December 2022 and September 2025.
- Proficiency in C++ and Python programming, applied to compiler projects
- Experience developing compiler optimizations or ML framework internals

- Knowledge of code generation, compute graph optimization, resource scheduling
- Experience optimizing Tensorflow, PyTorch or JAX deep learning models
- Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM
- Familiarity with CUDA programming for GPU acceleration