Sr. Machine Learning Engineer, AGI Foundations

The Artificial General Intelligence (AGI) team at Amazon is seeking a Senior Machine Learning Engineer to spearhead the development of cutting-edge multimodal systems and large language models. This role combines advanced machine learning expertise with software engineering excellence, focusing on scaling pre-training workflows and building efficient models that will directly impact customers. The position offers the opportunity to work with Amazon's vast data resources and computing infrastructure to push the boundaries of AGI development.

The ideal candidate will lead the technical direction for pre-training multimodal LLMs, working closely with applied scientists to optimize training workflows using distributed training techniques and frameworks like FSDP, NVIDIA NeMo, and Megatron Core. This role requires both technical depth in machine learning and the ability to drive system architecture decisions that enable scalable, high-performance solutions.

As part of the AGI Foundations team, you'll be at the forefront of building industry-leading generative AI models, working in an Agile environment with aggressive delivery schedules. The role offers competitive compensation ranging from $151,300 to $261,500 based on location, plus equity and comprehensive benefits.

The position demands a unique blend of machine learning expertise, software engineering skills, and leadership ability. You'll need to demonstrate strong communication skills, the ability to work effectively with both technical and business stakeholders, and a track record of delivering complex technical products in a fast-paced environment. This is an exceptional opportunity to shape the future of artificial general intelligence at one of the world's leading technology companies.