Senior ML 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 Senior ML Engineer, you'll join a multidisciplinary team of strategists, scientists, engineers, and architects working directly with customers across industries to fine-tune and deploy customized generative AI applications at scale.

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

Working closely with AWS's custom AI accelerators, you'll optimize models for high-performance deployment using AWS Inferentia and Trainium, leveraging the AWS Neuron SDK. This role offers unique opportunities to collaborate directly with enterprise customers and foundational model providers, understanding their challenges and co-developing tailored generative AI solutions.

As an Amazonian leader, you'll embody Amazon's Leadership Principles while coaching and mentoring others in best practices, performance, and career development. The position requires strong technical expertise combined with leadership skills, as you'll be expected to drive work rather than simply participate in it.

The role sits within AWS Professional Services, offering exposure to diverse customer challenges and the chance to work with cutting-edge AI technologies. You'll be part of a culture that values continuous learning, diversity, and work-life harmony. The team provides extensive knowledge-sharing opportunities, mentorship, and career advancement resources to help you develop as a well-rounded professional.

This position represents an exceptional opportunity to shape the future of AI technology while working with some of AWS's most important clients. If you're passionate about innovation in generative AI and want to lead transformative projects at scale, this role offers the perfect platform to make a significant impact.