Machine Learning Engineer, Generative AI Innovation Center

World's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
2+ years of experience
AI · Enterprise SaaS

Description For 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 who collaborate with customers across industries to fine-tune and deploy customized generative AI applications at scale.

Your role will focus on driving the development of custom Large Language Models (LLMs) across languages, domains, and modalities. You'll be responsible for fine-tuning state-of-the-art LLMs for diverse use cases while optimizing models for high-performance deployment on AWS's custom AI accelerators. This position offers a unique opportunity to innovate at the cutting edge of AI technology, working with massive-scale LLM training pipelines and delivering next-generation AI solutions for top AWS clients.

AWS values diverse experiences and work-life harmony, fostering an inclusive team culture through employee-led affinity groups and ongoing learning experiences. The company provides extensive knowledge-sharing opportunities, mentorship programs, and career advancement resources to help you develop as a well-rounded professional.

Working at AWS means joining the pioneer of cloud computing that continues to push the boundaries of innovation. You'll be part of a team that values curiosity and learning, with access to the latest technologies and opportunities to work on transformative AI projects. The role offers competitive compensation, comprehensive benefits, and the chance to make a significant impact in the field of generative AI while working with enterprise-scale customers and cutting-edge technology.

Last updated 16 minutes ago

Responsibilities For Machine Learning Engineer, Generative AI Innovation Center

  • Design and implement distributed training pipelines for LLMs using tools such as FSDP and DeepSpeed
  • Adapt LLMs for new languages, domains, and vision applications through pre-training and fine-tuning
  • Optimize AI models for deployment on AWS Inferentia and Trainium
  • Collaborate with enterprise customers and foundational model providers
  • Develop custom kernels for enhanced performance

Requirements For Machine Learning Engineer, Generative AI Innovation Center

Python
  • 2+ years of professional software development experience
  • Proficiency in at least one programming language
  • Hands-on experience with deep learning and machine learning methods
  • 2+ years of non-internship design or architecture experience
  • Hands-on experience with generative AI technology

Benefits For Machine Learning Engineer, Generative AI Innovation Center

Medical Insurance
Vision Insurance
Dental Insurance
  • Medical benefits
  • Work-life balance
  • Career growth opportunities
  • Mentorship programs
  • Learning and development resources

Interested in this job?

Jobs Related To Amazon Machine Learning Engineer, Generative AI Innovation Center

Machine Learning Engineer II, StoreGen

Machine Learning Engineer II position at Amazon's StoreGen team, focusing on AI-powered software development tools and practices with competitive compensation and benefits.

Amazon Q Delivery Engineer, Amazon Q Customer Success Team (Q-CST)

AWS Delivery Engineer position focusing on implementing Generative AI solutions using Amazon Q and Bedrock, combining technical expertise with customer success.

Machine Learning Engineer II, AWS Just-Walk-Out Science Team

Machine Learning Engineer role at Amazon's AWS Just-Walk-Out team, focusing on computer vision and deep learning for autonomous retail technology.

SDE-II, Alexa Sensitive Content & Intelligence

SDE-II position at Amazon's Alexa team focusing on content intelligence and trust, using AI/ML to protect users from sensitive content across all Alexa interactions.

Software Dev Engineer (ML), AGI Information - Knowledge Graphs

Machine Learning Engineer position at Amazon's AGI Information team, focusing on developing ML systems for knowledge graphs and LLM applications.