Startups · AI

Founding Machine Learning - World Models

One Robot · San Francisco · On-site

← All jobs
About One Robot

World models for robots. Backed by Y Combinator.

About the role

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.

What they're looking for

  • 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
More about this role

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.

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.

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.

Read the full posting on One Robot's site ↗

One Robot Founding Engineers

Build your edge while you search

Free tools for founders and investors, plus VC Unfiltered, our take on startups, venture and the people who build them.