Job Description

Join Amazon's Frontier AI & Robotics team as a Software Development Engineer, where you'll optimize large-scale transformer architectures for robotics applications. Collaborating with scientists, you'll transform research into high-performance production systems, leveraging your expertise in CUDA and TensorRT to achieve exceptional inference efficiency.

Requirements

3+ years of non-internship professional software development experience
2+ years of non-internship design or architecture of new and existing systems experience
Experience programming with at least one software programming language

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

Employees at Amazon are often offered comprehensive health benefits—including multiple medical plan options (no pre-existing condition exclusions, 100% covered in-network preventive care), dental and vision plans, a 24/7 medical advice line from day one, expert second-opinion services, and broad mental-health support with several free counseling sessions (including pediatric). Financial wellness typically includes a 401(k) with company match (up to 2%), Restricted Stock Units (equity), FSAs, an emergency savings program, product and partner discounts, and even college-savings and home-purchase programs. Overall, the package is designed to support employees and their families’ health, finances, and day-to-day life.

Is this job posting expired or no longer available?