Amazon Sr ML Compiler Engineer - Automated Reasoning Science Annapurna Labs 

United States, Washington, Seattle 

845488985

Key job responsibilitiesAs a Sr. ML Compiler Engineer III on the Neuron Compiler Automated Reasoning Group, you will develop and maintain tooling for best-in-class technology for raising the bar of the Neuron Compiler's accuracy and reliability. You will help lead the efforts building fuzzers and specification synthesis tooling for our LLVM-based compiler. You will work in a team with a science focus, and strive to push what we do to the edge of what is known, to best deliver our customers.Strong software development skills using C++/Python are critical to this role.A science background in compiler development is strongly preferred. A background in Machine Learning and AI accelerators is preferred, but not required.

- 6+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- 5+ years of experience in developing compiler features and optimizations
- Proficiency in C++ and Python programming, applied to compiler or verification projects
- Familiarity with LLVM, including knowledge of abstract interpretation and polyhedral domains
- Demonstrated scientific approach to software engineering problems

- Masters degree or PhD in computer science or equivalent
- Experience with deep learning frameworks like TensorFlow or PyTorch
- Understanding of large language model (LLM) training processes
- Knowledge of CUDA programming for GPU acceleration