Amazon PhD ML Compiler Engineer - Annapurna Labs 

United States, California, Cupertino 

785453400

You: As a PhD Machine Learning Compiler Engineer 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.

- To qualify, applicants should have earned (or will earn) a PhD between December 2023 and September 2025.
- Proficiency in C++ and Python programming, applied to compiler or verification projects
- Experience developing compiler optimizations or ML framework internals through research projects
- Understanding of systems design and distributed architecture principles

- Experience optimizing Tensorflow, PyTorch or JAX deep learning models
- Experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM
- Knowledge of CUDA programming for GPU acceleration