ML Compiler Engineer I
| Company | Amazon |
|---|---|
| Location | Seattle, WA, USA, Cupertino, CA, USA |
| Salary | $99500 – $200000 |
| Type | Full-Time |
| Degrees | Bachelor’s, Master’s |
| Experience Level | Entry Level/New Grad |
Requirements
- 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
Responsibilities
- Innovating and delivering creative SW Designs to develop new services, solve operational problems, drive improvements in developer velocity, or positively impact operational safety
- Writing requirements capturing documents, design documents, integration test plans, and deployment plans
- Communicating status and progress of deliverables to schedule, and sharing learnings/ innovations with your team and stakeholders
Preferred Qualifications
- 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