Machine Learning Engineer, Generative AI Innovation Center

Job Description

Join the Generative AI Innovation Center at AWS to drive the development of custom Large Language Models (LLMs) and deploy next-generation AI solutions for top AWS clients.

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

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)