Taro Logo

Machine Learning Engineer II, AWS Just-Walk-Out Science Team

A global technology company pioneering innovations in e-commerce, cloud computing, AI, and retail technology.
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Enterprise SaaS · Retail

Description For Machine Learning Engineer II, AWS Just-Walk-Out Science Team

Amazon's AWS Just Walk Out (JWO) team is revolutionizing retail with checkout-free shopping experiences powered by cutting-edge AI technology. As a Machine Learning Engineer II, you'll be part of the team developing autonomous AI agents that make intelligent decisions based on visual inputs and understand customer behavior patterns. The role combines advanced machine learning, computer vision, and distributed systems to create innovative solutions for the future of retail.

You'll work on designing and maintaining large-scale distributed training systems to support multi-modal foundation models for autonomous retailing. The position requires expertise in GPU optimization, deep learning frameworks, and the ability to scale complex ML models while maintaining high performance and reliability.

The team operates in a startup-like environment within Amazon, focusing on pushing the boundaries of computer vision, deep learning, and real-time systems. You'll collaborate with Applied Scientists and Engineers to develop and optimize training pipelines, integrate state-of-the-art models, and solve complex technical challenges.

This role offers significant opportunity for growth and impact, as you'll be working on technology that directly affects millions of customers' shopping experiences. The team values work-life balance, mentorship, and professional development, fostering an inclusive culture that embraces diversity and innovation.

The position combines technical expertise with business impact, as you'll be contributing to a revolutionary retail technology that's changing how people shop. You'll have access to Amazon's vast computing resources and data, enabling you to work on large-scale AI applications that push the boundaries of what's possible in retail technology.

Working at Amazon means being part of a culture that values innovation, customer obsession, and long-term thinking. The team offers flexibility in working hours, strong mentorship opportunities, and a collaborative environment where you can grow your career while working on cutting-edge technology that's reshaping the retail industry.

Last updated 5 minutes ago

Responsibilities For Machine Learning Engineer II, AWS Just-Walk-Out Science Team

  • Collaborate with Applied Scientists to integrate state-of-the-art model architectures
  • Process massive data and scale machine learning models
  • Design and maintain large-scale distributed training systems
  • Develop robust monitoring and debugging tools
  • Collaborate with Engineers and Applied Scientists to investigate design approaches
  • Lead development of novel algorithms and modeling techniques

Requirements For Machine Learning Engineer II, AWS Just-Walk-Out Science Team

Python
  • 3+ years of non-internship professional software development experience
  • 2+ years of design or architecture experience
  • Proficient in Python or related language
  • Hands-on model training experience in PyTorch and deep learning frameworks
  • Familiar with modern visual-language models and multi-modal AI systems
  • Proficient in training profilers and performance analysis tools

Benefits For Machine Learning Engineer II, AWS Just-Walk-Out Science Team

Medical Insurance
401k
Parental Leave
  • Flexible working hours
  • Work-life balance
  • Career growth opportunities
  • Mentorship
  • Comprehensive benefits package

Interested in this job?

Jobs Related To Amazon Machine Learning Engineer II, AWS Just-Walk-Out Science Team

Software Engineer - ML and Distributed Systems, Amazon Personalize

Senior Software Engineering role at AWS focusing on machine learning and distributed systems for Amazon Personalize, building scalable ML platforms and recommendation systems.

Software Engineer - ML and Distributed Systems, Amazon Personalize

Senior Software Engineering role at AWS focusing on machine learning and distributed systems for Amazon Personalize, building scalable recommendation systems.

Software Development Engineer, Alexa Customer Journeys

Software Development Engineer role at Amazon working on Alexa Customer Journeys, focusing on Generative AI and LLM technologies to enhance customer experiences.

Software Development Engineer II - DSO, Demand Science Optimization (DSO)

Software Development Engineer II position at Amazon's DSO team, focusing on ML-driven demand forecasting and supply optimization for Amazon Devices.

Software Development Engineer II - Machine Learning, Sponsored Products

SDE II role at Amazon Advertising focusing on machine learning systems for Sponsored Products, handling billions of daily ad requests with ML-driven relevancy matching.