Sr. Machine Learning Engineer

Job Description

Join an innovative team of engineers and scientists at Amazon to build foundational behavioral models for Amazon Stores using advanced AI techniques.

Requirements

• 5+ years of professional software development experience
• Experience in leading design or architecture of new systems
• Experience in machine learning pipelines and large language models
• Master's or PhD in Computer Sciences, Electrical Engineering, or Mathematics
• Experience with CUDA, Pytorch, or Tensorflow

Responsibilities

• Lead the design and implementation of a stable training infrastructure for large models
• Collaborate with other engineers to develop enhanced training infrastructure
• Learn cutting-edge technologies in Generative AI

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)