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 who collaborate with customers across industries to fine-tune and deploy customized generative AI applications at scale. The role involves working closely with foundational model providers to optimize AI models for Amazon Silicon, enhancing performance and efficiency.

Your responsibilities will span from designing large-scale training pipelines for LLMs to customizing models for specific use cases through techniques like RLHF. You'll be hands-on with state-of-the-art tools like FSDP and DeepSpeed, while also optimizing models for AWS's custom AI accelerators. The position offers a unique opportunity to work directly with enterprise customers, understanding their challenges and co-developing tailored AI solutions.

AWS values diverse experiences and maintains an inclusive culture through employee-led affinity groups and ongoing learning experiences. The team emphasizes work-life harmony and provides extensive resources for professional development, including mentorship and knowledge-sharing opportunities. As part of AWS, you'll be contributing to the world's most comprehensive cloud platform while working with cutting-edge AI technologies.

This role is perfect for someone who combines strong technical expertise in machine learning with a passion for solving complex business problems. You'll be part of a team that's shaping the future of AI applications while having the backing of AWS's robust infrastructure and resources.