Machine Learning Engineer, Generative AI Innovation Center

The Generative AI Innovation Center at AWS is at the forefront of empowering customers to leverage 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 who collaborate with customers across industries to fine-tune and deploy customized generative AI applications at scale.

Your role will focus on driving the development of custom Large Language Models (LLMs) across languages, domains, and modalities. You'll be responsible for designing and implementing distributed training pipelines using advanced tools like FSDP and DeepSpeed, ensuring scalability and efficiency. The position involves adapting LLMs through continued pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF).

Working closely with AWS's custom AI accelerators, you'll optimize models for high-performance deployment on AWS Inferentia and Trainium, leveraging the AWS Neuron SDK and developing custom kernels. A key aspect of the role involves direct interaction with enterprise customers and foundational model providers to understand their challenges and co-develop tailored generative AI solutions.

AWS values diverse experiences and work-life harmony, offering comprehensive benefits and professional development opportunities. You'll be part of AWS Global Services, working alongside technical experts across dozens of countries. The role offers competitive compensation ranging from $129,300 to $223,600 based on location and experience, plus additional benefits including equity and sign-on payments.

This position requires strong software development skills, deep learning expertise, and hands-on experience with generative AI technology. You'll have the opportunity to innovate at the cutting edge of AI while helping top AWS clients implement next-generation AI solutions. The collaborative environment, mentorship opportunities, and focus on continuous learning make this an ideal role for someone passionate about advancing the field of generative AI.