Sr. Research Engineer, Machine Learning, AGI Foundations

Amazon's Artificial General Intelligence (AGI) Foundations team is seeking an exceptional Senior Software Development Engineer with deep expertise in machine learning to spearhead the development of cutting-edge multimodal systems and Large Language Models (LLMs).

This role presents a unique opportunity to work at the forefront of AI innovation, where you'll lead the development of novel algorithms and modeling techniques that push the boundaries of what's possible with multimodal systems. You'll have access to Amazon's vast computational resources and diverse data sources, enabling you to accelerate development in multimodal LLMs and Generative AI.

As a senior engineer, you'll have significant influence on the team's technical direction and overall strategy. You'll work at the fascinating intersection of engineering and applied science, focusing on scaling pre-training workflows and building efficient models. Your responsibilities will include driving system architecture decisions and establishing best practices for quality infrastructure.

The role requires collaboration with applied scientists to optimize training workflows using advanced distributed training techniques and frameworks like FSDP, NVIDIA NeMo, and Megatron Core. You'll be working in an Agile/Scrum environment, delivering high-quality software against aggressive schedules.

The ideal candidate brings not only technical excellence but also strong leadership qualities. You'll need to demonstrate exceptional communication skills, as you'll be working with both technical and business partners. Your success in this role will directly impact Amazon's customers through the development of industry-leading AI models.

The position offers competitive compensation, including a base salary range of $151,300 to $261,500 depending on location, plus equity, sign-on bonuses, and comprehensive benefits. Join Amazon's AGI team to be part of shaping the future of artificial intelligence while working with some of the most advanced technologies in the field.