Tesla is seeking an AI Engineer to work on the Optimus Policy, focusing on the data, model training, evaluation, and deployment of policies for various tasks. The role involves creating scalable datasets, designing efficient policies, and deploying them on Optimus, contributing to the development of general embodied intelligence through reinforcement learning.

• 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 (vision, proprioception, tactile) and modern ML methods
• Production engineering mindset focused on safety-critical, low-latency systems
• Passion for building general embodied intelligence through reinforcement learning

• Create and scale datasets from a variety of sources (human, bot, simulation)
• Design and train efficient and performant policies that complete a wide variety of tasks on Optimus
• Create evaluation benchmarks for Optimus policies for fast iterative improvement
• Large-scale pre-training and reinforcement learning to create intelligent and robust policies
• Deploy policies on Optimus

• Employees at Tesla are often offered day-one coverage with multiple medical options (some at $0 paycheck cost), dental/vision, company HSA contributions, a 401(k) match, and equity programs. Most roles also include paid time off and holidays, family-building support, employee assistance, commuter and childcare benefits, and access to discounts and wellness programs.