Machine Learning Engineer, Gen AI Innovation Center, AWS

Job Description

The Generative AI Innovation Center at AWS empowers customers to harness state of the art AI technologies for transformative business opportunities. Join as a Software Development Engineer (SDE) to drive the development of custom Large Language Models (LLMs) and optimize AI models for high-performance deployment on AWS's custom AI accelerators.

Requirements

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture of new and existing systems experience
• Experience programming with at least one software programming language
• Hands-on experience with deep learning and/or machine learning methods
• 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 through continued pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF)
• 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)