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
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 LLM development and optimization. The role involves designing distributed training pipelines, customizing models for various domains, and optimizing performance on AWS Silicon. You'll collaborate directly with enterprise customers and model providers to develop tailored AI solutions.

The position offers unique opportunities to work with state-of-the-art AI technologies while solving real-world challenges at scale. You'll be responsible for implementing sophisticated training pipelines using tools like FSDP and DeepSpeed, fine-tuning models with techniques including RLHF, and optimizing deployments for AWS's custom AI accelerators.

AWS provides a collaborative environment focused on work-life harmony and inclusive culture. The team values diverse experiences and offers extensive opportunities for professional growth through mentorship, knowledge-sharing, and career development resources. You'll be part of AWS's mission to pioneer cloud computing innovation while working with top enterprise clients on next-generation AI solutions.

The role combines technical depth in machine learning with practical business impact, requiring both strong engineering skills and customer collaboration abilities. You'll have access to AWS's comprehensive cloud platform and custom hardware accelerators while working alongside experts in AI/ML, cloud architecture, and business strategy.

Last updated 21 minutes 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

Requirements For Machine Learning Engineer, Generative AI Innovation Center

Python
  • Experience programming with at least one software programming language
  • 3+ years of non-internship professional software development experience
  • 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
401k
Parental Leave
  • Work-life harmony
  • Inclusive team culture
  • Mentorship and career growth
  • Employee-led affinity groups
  • Knowledge-sharing opportunities

Interested in this job?

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

Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference

Senior Software Engineer role at AWS Neuron, focusing on developing and optimizing distributed ML inference solutions for cloud-scale AI accelerators.

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior ML Engineer role at AWS Neuron team, focusing on distributed training solutions for custom ML accelerators, offering competitive pay and comprehensive benefits.

Senior ML Engineer, Generative AI Innovation Center

Senior ML Engineer role at AWS's Generative AI Innovation Center, focusing on developing and optimizing custom LLMs while leading technical initiatives and mentoring team members.

ML Engineer, Generative AI Innovation Center

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

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior ML Engineer role at Amazon's Annapurna Labs, focusing on distributed training solutions for AWS Neuron platform and ML accelerators.