Machine Learning Engineer, Generative AI Innovation Center

Job Description

The SDE position at the Generative AI Innovation Center at AWS involves driving the development of custom Large Language Models (LLMs) across languages, domains, and modalities, fine-tuning state-of-the-art LLMs for diverse use cases, and optimizing models for high-performance deployment on AWS’s custom AI accelerators.

Requirements

• 2+ years of professional software development experience
• Proficiency in at least one programming language
• Hands-on experience with deep learning and machine learning methods
• 2+ years of non-internship design or architecture experience
• Hands-on experience with generative AI technology

Responsibilities

• Design and implement distributed training pipelines for LLMs
• Adapt LLMs for new languages, domains, and vision applications
• Optimize AI models for deployment on AWS Inferentia and Trainium
• Interact with enterprise customers and foundational model providers to co-develop tailored generative AI solutions

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)