Machine Learning Engineer, Generative AI Innovation Center

The Generative AI Innovation Center at AWS is at the forefront of empowering customers to leverage AI technologies for transformative business opportunities. As a Machine Learning Engineer, you'll join a multidisciplinary team of strategists, scientists, engineers, and architects working on cutting-edge generative AI applications. The role involves developing custom Large Language Models (LLMs), implementing distributed training pipelines, and optimizing models for AWS's custom AI accelerators.

You'll be responsible for fine-tuning state-of-the-art LLMs for diverse use cases while ensuring high-performance deployment. The position offers an exciting opportunity to work with enterprise customers and foundational model providers, solving complex technical challenges and developing tailored generative AI solutions. You'll use tools like Fully Sharded Data Parallel (FSDP) and DeepSpeed for large-scale training pipelines, and work with AWS Neuron SDK for model optimization.

AWS offers a comprehensive benefits package, including medical coverage and 401k, along with a strong emphasis on work-life harmony. The company values diverse experiences and fosters an inclusive culture through employee-led affinity groups and ongoing learning opportunities. You'll have access to mentorship and career growth resources, working in an environment that encourages innovation and continuous learning.

The role requires at least 2 years of professional software development experience, with hands-on experience in machine learning and generative AI. You'll be part of AWS Professional Services, working in one of several major tech hubs across the United States. The compensation is competitive, ranging from $129,300 to $223,600 based on location and experience, plus additional benefits and equity opportunities.