PhD] ML Compiler Engineer – Annapurna Labs – Annapurna Labs
| Company | Amazon |
|---|---|
| Location | Seattle, WA, USA, Toronto, ON, Canada, Cupertino, CA, USA |
| Salary | $129300 – $223600 |
| Type | Full-Time |
| Degrees | PhD |
| Experience Level | Entry Level/New Grad |
Requirements
- 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
Responsibilities
- Support the ground-up development and scaling of a compiler to handle the world’s largest ML workloads
- Architect and implement business-critical features
- Publish cutting-edge research
- Contribute to a brilliant team of experienced engineers
- Leverage technical communications skills as a hands-on partner to AWS ML services teams
- Involved in pre-silicon design, bringing new products/features to market, and other exciting projects
Preferred Qualifications
- 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