Amazon Customer Services Shipping and Delivery Support (SDS) provides support to drivers who deliver packages to customers on behalf of Amazon. We are seeking a passionate and talented Machine Learning Engineer (MLE) to join our team in advancing state-of-the-art Generative AI technologies and Large Language Models (LLMs). In this role you will design develop and deploy highly scalable AI models that benefit drivers customer service associates and customers. You will work closely with applied scientists and research teams to productionalize Machine Learning (ML) models focusing on novel LLM training techniques optimizations and customization capabilities such as fine-tuning and distillation.
Key job responsibilities
- Collaborate with Applied Scientists to integrate state-of-the-art model architectures into the training pipeline integrate state-of-the-art ML/LLM into CS system.
- Collaborate with Applied Scientists to process massive data scale LLM while optimizing GPU utilization memory management and the training workflows (like offloading optimizer states massive parallelization etc).
- Cross-team collaboration to facilitate deployment and test of models in the production system.
- Design and maintain large-scale distributed training systems to support multi-modal foundation models.
- Optimize GPU utilization for efficient model training and fine-tuning on massive datasets.
- Develop robust monitoring and debugging tools to ensure the reliability and performance of training workflows support piloting the LLMs and identify the related issues in the system.
- Collaborate with Engineers and Applied Scientists to investigate design approaches prototype new LLM technique and evaluate technical feasibility identify and solve complex problems.
A day in the life
If you are not sure that every qualification on the list above describes you exactly wed still love to hear from you! At Amazon we value people with unique backgrounds experiences and skillsets. If youre passionate about this role and want to make an impact on a global scale please apply!
Amazon offers a full range of benefits that support you and eligible family members including domestic partners and their children. Benefits can vary by location the number of regularly scheduled hours you work length of employment and job status such as seasonal or temporary employment. The benefits that generally apply to regular full-time employees include:
1. Medical Dental and Vision Coverage
2. Maternity and Parental Leave Options
3. Paid Time Off (PTO)
4. 401(k) Plan
About the team
The Shipping Delivery and Support (SDS) Tech teams mission is to enable successful deliveries through fast and efficient support experiences for our drivers recipients and associates. The science team utilizes advanced scientific methods including LLM to support the business improve driver support quality and become more efficient with deliveries.
- 3 years of non-internship professional software development experience
- 2 years of non-internship design or architecture (design patterns reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language
- 1 years of machine learning statistical modeling data mining and analytics techniques experience
- 3 years of full software development life cycle including coding standards code reviews source control management build processes testing and operations experience
- Bachelors degree in computer science or equivalent
- Experience in machine learning data mining information retrieval statistics or natural language processing
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status disability or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process including support for the interview or onboarding process please visit
for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $129300/year in our lowest geographic market up to $223600/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge skills and experience. Amazon is a total compensation company. Dependent on the position offered equity sign-on payments and other forms of compensation may be provided as part of a total compensation package in addition to a full range of medical financial and/or other benefits. For more information please visit This position will remain posted until filled. Applicants should apply via our internal or external career site.