Sr. Software Development Engineer, FAR (Frontier AI & Robotics)

Job Description

Join Amazon's Frontier AI & Robotics team as a Senior Machine Learning Engineer to work on transforming cutting-edge research into high-performance production systems.

Requirements

• Bachelor's degree in computer science or equivalent
• 5+ years of non-internship professional software development experience
• 5+ years of programming with at least one software programming language experience
• 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems 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
• Track record of optimizing ML models for production

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
• Maintain high engineering standards through proper testing, documentation, and code review practices

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)