AI Engineer, Policy, Optimus

Tesla is scaling manufacturing of Optimus bots and seeks an AI Engineer to work on policy training, evaluation, and deployment. The role involves creating datasets, designing policies, and large-scale reinforcement learning for real-world robotics tasks.

• Create and scale datasets from human, bot, and simulation sources.
• Design and train efficient policies for a wide variety of tasks on Optimus.
• Create evaluation benchmarks for fast iterative improvement.
• Large-scale pre-training and reinforcement learning.
• Deploy policies on Optimus.

• Hands-on experience training and deploying policies for robotics or autonomous systems.
• Strong proficiency in PyTorch and large-scale distributed training.
• Proven track record shipping real-world neural policies.
• Familiarity with multimodal sensing and modern ML methods.
• Production engineering mindset for safety-critical systems.
• Passion for building general embodied intelligence through reinforcement learning.