Machine Learning Engineer, Generative AI Innovation Center

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
$129,300 - $223,600
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
3+ 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 working on cutting-edge Large Language Models (LLMs) and their applications. The role involves designing and implementing distributed training pipelines, fine-tuning state-of-the-art LLMs, and optimizing models for AWS's custom AI accelerators.

You'll be responsible for adapting LLMs for various languages, domains, and vision applications through continued pre-training, fine-tuning, and Reinforcement Learning with Human Feedback (RLHF). Working directly with enterprise customers and foundational model providers, you'll help solve complex business and technical challenges while developing tailored generative AI solutions.

AWS offers a comprehensive compensation package, including competitive base pay ranging from $129,300 to $223,600 depending on location, plus equity and sign-on bonuses. The company values work-life harmony and fosters an inclusive culture through employee-led affinity groups and ongoing learning experiences.

The ideal candidate brings 3+ years of software development experience, strong expertise in machine learning and deep learning methods, and hands-on experience with generative AI technology. You'll be part of AWS's mission to pioneer cloud computing and continue pushing the boundaries of innovation in AI technology.

This role offers unique opportunities to work with cutting-edge AI technologies, collaborate with top AWS clients, and make significant contributions to the future of generative AI while being supported by a culture that values diversity, continuous learning, and professional growth.

Last updated 3 days ago

Responsibilities For Machine Learning Engineer, Generative AI Innovation Center

  • Design and implement distributed training pipelines for LLMs
  • Adapt LLMs for new languages, domains, and vision applications
  • Optimize AI models for deployment on AWS Inferentia and Trainium
  • Interact with enterprise customers and foundational model providers
  • Develop custom kernels for enhanced performance
  • Fine-tune state-of-the-art LLMs for diverse use cases

Requirements For Machine Learning Engineer, Generative AI Innovation Center

Python
  • 3+ years of non-internship professional software development experience
  • Experience programming with at least one software 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
  • Bachelor's degree in computer science or equivalent (preferred)
  • Experience with ML library or framework (preferred)
  • 2+ years of experience in full software development life cycle (preferred)

Benefits For Machine Learning Engineer, Generative AI Innovation Center

Medical Insurance
Equity
  • Medical, financial, and other benefits
  • Work-life harmony
  • Mentorship and career growth opportunities
  • Inclusive team culture
  • Employee-led affinity groups
  • Equity compensation
  • Sign-on payments

Interested in this job?

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

Senior ML Engineer, GenAIID

Senior ML Engineer position at AWS's Generative AI Innovation Center, focusing on developing and optimizing Large Language Models for enterprise solutions.

Machine Learning Engineer, AWS Generative AI Innovation Center

Senior ML Engineer role at AWS Generative AI Innovation Center, focusing on developing and implementing generative AI solutions for enterprise customers.

ML Engineer, GenAIID

Senior ML Engineer position at AWS's Generative AI Innovation Center, focusing on LLM development, model optimization, and enterprise AI solutions.

Machine Learning Engineer, AI/LLM, Amazon Shopping

Senior ML Engineer role at Amazon focusing on LLM-based systems, offering competitive pay and benefits in Seattle.

Software Development Engineer AI/ML

Senior Software Engineer role at Amazon focusing on AI/ML development for revolutionary shopping experiences, offering competitive compensation and growth opportunities.