Machine Learning Engineer, Generative AI Innovation Center

The Generative AI Innovation Center at AWS is seeking a Machine Learning Engineer to join their cutting-edge team. This role focuses on empowering customers to leverage state-of-the-art AI technologies for transformative business opportunities. As part of a multidisciplinary team of strategists, scientists, engineers, and architects, you'll be responsible for developing custom Large Language Models (LLMs) across various domains and modalities.

The position involves working with enterprise-level customers and foundational model providers to optimize AI models for Amazon Silicon, enhancing both performance and efficiency. You'll be at the forefront of AI innovation, managing end-to-end LLM training pipelines at massive scale and delivering next-generation AI solutions for top AWS clients.

Key responsibilities include designing distributed training pipelines using advanced tools like FSDP and DeepSpeed, customizing LLMs through pre-training and fine-tuning, and optimizing models for AWS Inferentia and Trainium. The role requires strong software development skills, deep learning expertise, and hands-on experience with generative AI technologies.

AWS offers an inclusive culture with employee-led affinity groups, mentorship opportunities, and continuous learning experiences. The company values work-life harmony and provides resources for career advancement, including AWS certifications and collaboration with industry experts. This position offers the chance to work with cutting-edge AI technology while contributing to AWS's mission of being Earth's Best Employer.