Machine Learning Engineer, Generative AI Innovation Center - Model Customization

The Generative AI Innovation Center at AWS is at the forefront of AI technology, empowering customers to leverage cutting-edge AI solutions for transformative business opportunities. As a Machine Learning Engineer in this role, you'll be part of a multidisciplinary team working on state-of-the-art Large Language Models (LLMs).

The position offers a unique opportunity to work with enterprise-scale AI applications, focusing on model customization, fine-tuning, and optimization for AWS's custom AI accelerators. You'll be responsible for building distributed training pipelines using advanced tools like Fully Sharded Data Parallel (FSDP) and DeepSpeed, ensuring both scalability and efficiency.

What makes this role particularly exciting is the chance to work directly with top AWS clients and foundational model providers, helping shape the future of AI applications across various industries. You'll be involved in adapting LLMs for multiple languages and domains, implementing Reinforcement Learning with Human Feedback (RLHF), and optimizing models for AWS's specialized hardware.

AWS offers a comprehensive benefits package, strong emphasis on work-life harmony, and a culture that values diversity and inclusion. The company provides extensive opportunities for professional growth through mentorship programs and continuous learning initiatives. As part of AWS, you'll be working with the world's leading cloud platform provider, contributing to innovations that impact businesses globally.

The role requires a strong background in software development and machine learning, with hands-on experience in generative AI technologies. You'll be joining a team that values curiosity, learning, and embraces diverse experiences and perspectives. The position offers competitive compensation, including equity opportunities and comprehensive benefits, reflecting AWS's commitment to attracting and retaining top talent.