Machine Learning Engineer, Amazon General Intelligence (AGI)

The Artificial General Intelligence (AGI) team at Amazon is seeking a talented Machine Learning Engineer to spearhead the development of cutting-edge multi-modal and multi-lingual Large Language Models (LLM). This role offers an exciting opportunity to work at the forefront of AI technology, leveraging Amazon's vast computing resources and diverse data sources.

As an MLE on the AGI team, you'll be responsible for developing training algorithms and optimizing performance of state-of-the-art large model training using NVIDIA GPUs and AWS Trainium. You'll work on implementing efficient memory management techniques, kernel fusion, and custom kernels to improve GPU utilization. The role involves collaborating with applied scientists to process massive datasets and scale machine learning models effectively.

The position offers the chance to make a significant impact on Amazon's AI initiatives, working with top minds in the field of deep learning and reinforcement learning. You'll be part of a team that's pushing the boundaries of what's possible in artificial intelligence, developing systems that revolutionize machine perception, interpretation, and interaction with humans and the physical world.

The compensation package is competitive, ranging from $129,300 to $223,600 based on location, plus potential equity, sign-on bonuses, and comprehensive benefits. This is an opportunity to work on groundbreaking technology while building your career at one of the world's leading tech companies.

The ideal candidate will have strong software development experience, expertise in GPU optimization, and a passion for artificial intelligence. You'll work in an Agile/Scrum environment, collaborating with cross-functional teams to deliver high-quality software solutions that will shape the future of AGI at Amazon.