• You love pursuing novel ideas and making them real, are a great communicator, and enjoy working with a highly capable team
• You are a fast learner and interested in reading and adapting the latest techniques and models in the emerging AI landscape to new applications
• You are a creative engineer who builds systems from the ground up
• PhD in machine learning, computer science, math or related field, or equivalent experience
• 5+ years experience developing and applying AI/ML methods in industry
• Experience developing for and operating physical robots
• Publications in top machine learning conferences (ICML, NeurIPS, CoRL etc)
• A track record of coming up with new ideas or improving upon existing ideas in machine learning or robotics
• Proficiency with Python and extensive experience with hardware accelerated tensor libraries such as PyTorch or JAX

Desirable

• As a Senior/Staff AI Engineer, you will be a team leader, working with experienced hardware and software engineers to develop cutting-edge autonomous capabilities for our humanoid robots
• You will focus on areas such as training foundational visuomotor policies for manipulation and locomotion, planning through learned embeddings of goals, tasks and environments and leveraging text and image foundation models for multi-modal conditioning
• You will utilize supervised, unsupervised and reinforcement learning methods on both simulation and hardware data
• Team leadership: roughly 50% individual contributor, 50% managing a highly capable team
• Stay up to date with the state of the art in reinforcement learning, imitation learning and generative models
• Apply, implement and improve state of the art machine learning algorithms and models and test them in real-world environments and hardware
• Own and pursue a research agenda, aligned with the innovation leadership, picking the most promising and impactful problems to work on
• Train models on collocated and cloud compute clusters with data from a variety of sources
• Full-time role, 40 hrs / week