Founding Machine Learning - World Models

One Robot · San Francisco, CA, US · In-office · Posted 11d ago

$150k - $275k

World models for robot evals and training.

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.

**What you'll do:**

* **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.

**Requirements:**

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

Apply on One Robot's site