Finding the best job has never been easier
Share
Work/Life Balance
Mentorship and Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Diverse Experiences
Amazon values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
- PhD in CS, ML or related field, or Master with equivalent years of experience
- Experience programming in Java, C++, Python or related language
- Experience developing ML models and systems to deliver product capabilities, and demonstrated ability to own problems end-to-end (able to innovate to bridge gaps and willing to learn missing knowledge)
- Patents or publications at top-tier peer-reviewed conferences or journals
- Prior experience with one or more model inference optimization techniques (quantization, distillation) and fluent in developing inference optimization techniques in frameworks like PyTorch and CUDA.
- Experience with large scale machine learning systems such as profiling and debugging and understanding of system performance and scalability.
- Prior experience in the training and fine tuning of Large Language Models (LLMs), or experience with various inference engines.
- Deep understanding of GPU architectures and experience optimizing for different hardware platforms and systems preferred.
These jobs might be a good fit