Taro Logo

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 working directly 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 various domains and modalities. You'll be responsible for implementing distributed training pipelines, fine-tuning state-of-the-art LLMs, and optimizing models for high-performance deployment on AWS's custom AI accelerators. This position offers a unique opportunity to work with cutting-edge AI technology while directly impacting AWS's top clients.

The role combines technical expertise in machine learning with practical business application, requiring both strong programming skills and the ability to collaborate with enterprise customers. You'll work closely with foundational model providers to optimize AI models for Amazon Silicon, enhancing both performance and efficiency. The position offers significant growth potential within AWS's innovative culture, with access to mentorship, knowledge-sharing, and career advancement resources.

AWS values diverse experiences and maintains an inclusive team culture, supported by employee-led affinity groups and ongoing learning experiences. The company emphasizes work-life harmony and provides comprehensive benefits including medical coverage and career development opportunities. This role is ideal for someone passionate about AI innovation who wants to work at the intersection of cutting-edge technology and practical business solutions.

Last updated 15 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
  • 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
401k
  • Comprehensive medical benefits
  • Work-life harmony
  • Career advancement resources
  • Mentorship opportunities
  • Inclusive team culture

Interested in this job?

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

Software Development Engineer II, ML_AI

AWS SageMaker AI seeks SDE II to build next-gen AI platforms for deep learning, focusing on LLMs and distributed ML systems with 100B+ parameter models.

Machine Learning Engineer, CreativeX

Machine Learning Engineer position at Amazon's CreativeX team, focusing on real-time ad personalization using advanced AI technologies including LLMs and computer vision.

Machine Learning Engineer, MLE II, QuickSight

Machine Learning Engineer II position at Amazon QuickSight, focusing on NLP and LLM applications for business intelligence, offering competitive compensation and comprehensive benefits.

Software Development Engineer II - DSO, Devices Forecasting

Software Development Engineer II position at Amazon's DSO team, focusing on ML-driven demand forecasting for Amazon devices, offering competitive compensation and opportunity to work with cutting-edge technology.

Machine Learning Engineer

Machine Learning Engineer role at Amazon's AGI team, focusing on developing multi-modal LLMs and Generative AI applications using cutting-edge technology and large-scale computing resources.