Machine Learning Engineer II, Shopping Conversation Foundation

Global technology company leading in e-commerce, cloud computing, and artificial intelligence
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Enterprise SaaS · E-Commerce

Description For Machine Learning Engineer II, Shopping Conversation Foundation

Amazon's Shopping Convo Foundations team is seeking a Machine Learning Engineer to lead the development of cutting-edge AI technology using Large Language Models and Natural Language Processing. This role focuses on building core foundational capabilities to train and optimize large language models, creating tools and infrastructure to enhance experiences like Rufus. The position offers an opportunity to work with state-of-the-art deep learning and generative models, developing internet-scale data solutions for customer-facing shopping conversation experiences.

The role combines technical expertise in machine learning with practical software engineering, requiring skills in distributed systems, model optimization, and large-scale data processing. You'll work alongside talented engineers and scientists, tackling unprecedented technical challenges while building experiences used by millions globally. The team values collaboration, diversity, and continuous learning, making it an ideal environment for those passionate about advancing the field of generative AI.

Key technical aspects include implementing distributed training pipelines using tools like FSDP and DeepSpeed, customizing LLMs through pre-training and RLHF, and optimizing models for AWS silicon. The position offers competitive compensation ($129,300-$223,600 based on location), comprehensive benefits, and the opportunity to work with cutting-edge technology at one of the world's leading tech companies.

This role is perfect for candidates who combine strong machine learning expertise with software engineering skills and are excited about creating innovative customer experiences. You'll have the chance to influence technical strategy, lead architecture decisions, and work with multiple teams across Amazon, making a significant impact on the future of AI-powered shopping experiences.

Last updated 3 days ago

Responsibilities For Machine Learning Engineer II, Shopping Conversation Foundation

  • Develop and maintain key services for evaluating and deploying large language models
  • 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
  • Work with peers to investigate design approaches and prototype new technology
  • Process massive data and scale machine learning models
  • Lead and influence overall tech strategy
  • Lead system architecture and spearhead best practices
  • Work in an Agile/Scrum environment
  • Collaborate with applied scientists and foundational model providers

Requirements For Machine Learning Engineer II, Shopping Conversation Foundation

Python
  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture experience
  • Experience programming with at least one software programming language
  • Hands-on experience with deep learning and/or machine learning methods
  • Bachelor's degree in computer science or equivalent (preferred)
  • 1+ years of experience with developing, deploying, or optimizing machine learning models (preferred)

Benefits For Machine Learning Engineer II, Shopping Conversation Foundation

Medical Insurance
Equity
  • Medical benefits
  • Financial benefits
  • Equity compensation
  • Sign-on payments
  • Comprehensive benefits package

Interested in this job?

Jobs Related To Amazon Machine Learning Engineer II, Shopping Conversation Foundation

Software Development Engineer, Finance Technology

Build AI/ML applications for Amazon's finance systems, focusing on data processing, forecasting, and automation within the FinTech team.

Machine Learning Engineer, MLE II, Amazon Q in QuickSight

Machine Learning Engineer role at Amazon working on Q in QuickSight, focusing on LLMs and NLP for business intelligence solutions.

Software Development Engineer, SageMaker Training Platform

Software Development Engineer position at Amazon's SageMaker Training Platform, building and scaling machine learning infrastructure.

Software Development Engineer | Amazon Delivery Experience (DEX) | Machine Learning, DEX AI Regionalization Engineering

Software Development Engineer position at Amazon's DEX team focusing on Machine Learning and delivery experience optimization

Software Development Engineer | Amazon Delivery Experience (DEX), DEX AI Regionalization Engineering

Senior Software Engineer role at Amazon's DEX team, focusing on ML-based delivery experience optimization with competitive compensation and benefits.