Taro Logo

Senior Machine Learning Engineer, Model Customization, Generative AI Innovation Center

AWS is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
$151,300 - $261,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Job Description

The Generative AI Innovation Center at AWS is at the forefront of empowering customers to leverage cutting-edge AI technologies for transformative business opportunities. As a Senior 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 various languages, domains, and modalities. You'll be responsible for implementing sophisticated training pipelines using advanced tools like FSDP and DeepSpeed, while also optimizing models for AWS's custom AI accelerators.

The position offers unique opportunities to work with enterprise customers and foundational model providers, helping them overcome technical challenges and co-develop tailored generative AI solutions. You'll be involved in the full lifecycle of LLM development, from fine-tuning and optimization to deployment at scale.

AWS values diverse experiences and work-life harmony, offering a collaborative environment where innovation thrives. You'll have access to extensive knowledge-sharing resources, mentorship opportunities, and career advancement pathways. The team culture promotes inclusion through employee-led affinity groups and inclusion events that foster stronger collaboration.

The role combines technical expertise with customer interaction, requiring both deep technical knowledge in machine learning and the ability to understand and address business challenges. You'll be working with state-of-the-art technology while having the opportunity to shape the future of AI applications across various industries.

This position is ideal for someone who is passionate about AI innovation, has strong technical skills in machine learning, and wants to work on impactful projects that help customers leverage the full potential of generative AI technologies.

Last updated 5 days ago

Responsibilities For Senior Machine Learning Engineer, Model Customization, Generative AI Innovation Center

  • Design and implement distributed training pipelines for LLMs using tools such as Fully Sharded Data Parallel (FSDP) and DeepSpeed
  • Adapt LLMs for new languages, domains, and vision applications through continued pre-training, fine-tuning, and RLHF
  • 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 Senior Machine Learning Engineer, Model Customization, Generative AI Innovation Center

Python
  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language
  • 5+ years of leading design or architecture of new and existing systems
  • Experience as a mentor, tech lead or leading an engineering team
  • Hands-on experience with deep learning and machine learning methods
  • Experience with design, development, and optimization of generative AI solutions

Benefits For Senior Machine Learning Engineer, Model Customization, Generative AI Innovation Center

Medical Insurance
Vision Insurance
Dental Insurance
  • Medical benefits
  • Work-life balance
  • Career growth opportunities
  • Mentorship programs
  • Training and certification opportunities