ML Infrastructure Engineer, Fauna

Amazon is seeking a Machine Learning Engineer to work directly alongside research scientists at Fauna Robotics. The role involves training, evaluating, and deploying models for robotic systems, focusing on hands-on machine learning applications and infrastructure management.

Responsibilities:

• Train and iterate on neural network policies for locomotion, manipulation, navigation, and perception using reinforcement and supervised learning
• Design and run experiments in simulation (Isaac Lab, MuJoCo, or similar) and transfer results to physical hardware
• Debug training runs end-to-end: diagnosing convergence failures, reward shaping issues, data quality problems, and sim-to-real gaps
• Optimize models for deployment on edge hardware (NVIDIA Jetson) with strict latency and memory constraints
• Build and maintain MLOps infrastructure: experiment tracking, model versioning, evaluation pipelines, and reproducible training workflows