Post your job offer for free on H1BConnect with no upfront cost!

Logo

Hire with Us
Amazon.com, Inc. logo

Machine Learning Engineer III, FAR (Frontier AI & Robotics)

Amazon.com, Inc.

4/4/2025

USA (Multiple Locations)

Full-time

Salary: $151,300 - $261,500 per year


Job Description

Join the next revolution in robotics at Amazon's Frontier AI & Robotics team as a Senior Machine Learning Engineer.

Requirements

  • Bachelor's degree in computer science or equivalent
  • 5+ years of non-internship professional software development experience
  • Experience as a mentor, tech lead or leading an engineering team
  • Strong expertise in Python, C++ and CUDA programming
  • Experience with TensorRT or similar ML optimization frameworks

Responsibilities

  • Drive inference optimization strategies for large-scale foundation models using TensorRT, CUDA, and other NVIDIA tools
  • Collaborate closely with scientists to influence model architectures for optimal hardware utilization
  • Design and implement efficient compilation pipelines for complex transformer architectures
  • Develop comprehensive benchmarking frameworks to measure and optimize model performance
  • Build robust monitoring solutions to ensure reliable model serving at scale
  • Explore and evaluate emerging optimization techniques including ONNX Runtime and other ML compilers

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)
Logo

© 2024 H1BConnect. All rights reserved.

Check out our sister site LatamDev for tech jobs in Latin America! 🌎