Machine Learning Engineer, Model Customization, Generative AI Innovation Center

Job :| Amazon Web Services, Inc.

The Generative AI Innovation Center at AWS empowers customers to harness state‑of‑the‑art AI technologies for transformative business opportunities. Our multidisciplinary team of strategists, scientists, engineers, and architects collaborates with customers across industries to fine‑tune and deploy customized generative AI applications at scale, and works closely with foundational model providers to optimize AI models for Amazon Silicon.

Key Job Responsibilities

• Design and implement distributed training pipelines for large‑scale language models using tools such as Fully Sharded Data Parallel (FSDP) and Deep Speed, ensuring scalability and efficiency.

• Adapt LLMs for new languages, domains, and vision applications through continued pre‑training, fine‑tuning, and Reinforcement Learning with Human Feedback (RLHF).

• Optimize AI models for deployment on AWS Inferentia and Trainium, leveraging the AWS Neuron SDK and developing custom kernels for enhanced performance.

• Interact with enterprise customers and foundational model providers to understand business and technical challenges, co‑developing tailored generative AI solutions.

About the Team

Our team values diverse experiences, curiosity, and connection. Together we innovate at AWS’s Generative AI Innovation Center and support customers worldwide in building, operating, and securing sophisticated AI solutions.

Basic Qualifications

• 3+ years of non‑internship professional software development experience.

• 2+ years of non‑internship design or architecture experience for new and existing systems (design patterns, reliability, and scaling).

• Bachelor’s degree in computer science or equivalent.

• Hands‑on experience with deep learning and/or machine learning methods (e.g., training, fine‑tuning, inference).

• Hands‑on experience with generative AI technology.

Preferred Qualifications

• 3+ years of full software development life cycle experience, including coding standards, code reviews, source control, build processes, testing, and operations.

• 1+ year of programming with at least one software programming language.

• 1+ year of experience deploying, optimizing, or developing machine learning models using a recognized ML library or framework.

Compensation

The base pay for this position ranges from $129,300/year in the lowest geographic market up to $223,600/year in the highest geographic market. Pay is based on market location and may vary depending on job‑related knowledge, skills, and experience. Additional compensation may include equity, sign‑on payments, and other forms of total compensation.

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, please visit  for more information.

#JLjbffr