AI Research Engineer - World Models

We build humanoid robots that work alongside people to solve labor shortages and create abundance.

• Build full-stack systems across data engineering, model architecture, and product delivery

• Develop high-throughput data loaders for large-scale, multi-modal robot datasets

• Implement tokenizers and transformers designed for web-scale robotic data

• Improve robot autonomy by advancing world model architectures

• Predict robot performance in the real world based on pre-training metrics such as log loss

• Strong experience with Python and related tools such as Bazel

• Proficiency with PyTorch or equivalent deep learning frameworks

• Familiarity with simulation platforms such as Isaac Sim or MuJoCo

• Experience working with multi-modal generative models combining video, audio, text, and action prediction

• Ability to design and optimize large-scale data pipelines for training

• Understanding of scaling laws and performance metrics for foundation models