Founding Machine Learning - World Models

One Robot • San Francisco, CA

6d • $150,000 - $275,000

About The Position

We build world models that simulate manipulation scenes faithfully enough to validate, and one day, train policies without touching a robot. You'll develop generative models that make this work, with the controllability and physical fidelity to match real-robot behavior.

Requirements

• Very strong coding in Python and PyTorch (or similar).
• Deep experience training image or video generation models end-to-end.
• Track record operating training runs at cluster scale.
• Working knowledge of multi-view geometry, scene reconstruction, and physical priors.

Responsibilities

• Train video and dynamics models: Develop world models with action conditioning for manipulation policies.
• Push long-horizon coherence: Develop architectures and training methods that extend rollout quality on hard physical tasks.
• Own training infrastructure: Run multi-GPU clusters, write custom CUDA, debug at scale.
• Build the world-model data engine: Design, implement, and improve a data engine that allows the world model to compound learning across customers and manipulation tasks.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed