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

Job Description

Join the AWS Solutions team to work on Just Walk Out Technology and innovate in computer vision, deep learning, and autonomous AI agents for dynamic retail environments.

Requirements

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship 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

Responsibilities

• Collaborate with Applied Scientists to integrate model architectures and MLLMs into training pipelines
• Design and maintain large-scale distributed training systems for autonomous retailing
• Develop monitoring and debugging tools for training workflows on large GPU clusters
• Investigate design approaches, prototype new technology, and solve complex problems

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)