Software Development Engineer, Frontier AI & Robotics

Join Amazon's Frontier AI & Robotics team to revolutionize robotics through breakthrough foundation models at production scale. As a Software Development Engineer embedded in the science team, you'll transform cutting-edge research into high-performance production systems. The role focuses on optimizing large-scale transformer architectures for robotics applications, requiring expertise in CUDA and TensorRT to achieve unprecedented inference efficiency at Amazon scale.

You'll work at the intersection of large language models and real-world robotics applications, balancing deep technical optimization work with strategic input on model architecture decisions. The team is building the future of intelligent robotics through frontier foundation models and end-to-end learned systems, tackling challenging problems in AI and robotics from sophisticated perception systems to adaptive manipulation strategies.

The position offers a unique combination of ambitious research vision and practical impact, leveraging Amazon's massive computational infrastructure and rich real-world datasets. You'll collaborate directly with world-class researchers while seeing your innovations deployed at unprecedented scale. The work spans the full spectrum of robotics intelligence, from multimodal perception using images, videos, and sensor data, to sophisticated manipulation strategies for diverse real-world scenarios.

Key responsibilities include optimizing transformer blocks using custom CUDA kernels, partnering with scientists to analyze and improve model architectures, implementing benchmarking frameworks, debugging performance bottlenecks, and maintaining robust monitoring systems. The role requires strong software engineering fundamentals combined with specialized knowledge in ML optimization and hardware acceleration.

The compensation reflects geographic market differences, ranging from $129,300 to $223,600 base pay, plus equity and comprehensive benefits. This is an opportunity to push the boundaries of what's possible in robotics while working with cutting-edge technology at one of the world's leading tech companies.