Machine Learning Engineer, Generative AI Innovation Center - Model Customization

The Generative AI Innovation Center at AWS is at the forefront of empowering customers with cutting-edge AI technologies for transformative business opportunities. As a Machine Learning Engineer, you'll join a multidisciplinary team of strategists, scientists, engineers, and architects working on state-of-the-art Large Language Models (LLMs). Your role will involve designing and implementing distributed training pipelines, customizing LLMs for various applications, and optimizing models for AWS's custom AI accelerators.

The position offers a unique opportunity to work with enterprise customers and foundational model providers, solving complex technical challenges and developing tailored generative AI solutions. You'll be responsible for implementing advanced techniques like Fully Sharded Data Parallel (FSDP), DeepSpeed, and Reinforcement Learning with Human Feedback (RLHF).

AWS values diverse experiences and maintains an inclusive culture where employee-led affinity groups foster a supportive environment. The company offers comprehensive benefits, mentorship opportunities, and a strong focus on work-life harmony. As part of AWS, you'll contribute to the world's most comprehensive cloud platform while working with cutting-edge AI technologies.

The role requires strong software development experience, deep learning expertise, and the ability to collaborate with customers. You'll be part of AWS's mission to pioneer cloud computing and continue pushing the boundaries of what's possible with AI technology. The position offers competitive compensation based on location and experience, along with equity opportunities and comprehensive benefits.