Founding ML Engineer: World Models & Simulation

One Robot (YC W26); San Francisco (CA); On-site; USD 150,000 - 275,000; Full time

Job summary

One Robot (YC W26) seeks an engineer to build world models that simulate manipulation scenes and validate policies in a scalable way. You will develop generative models with controllability and physical fidelity matching real-robot behavior.

You will train video and dynamics models on multi-GPU clusters, own data pipelines, and ensure learning across customers and tasks. Strong Python/PyTorch skills and video generation and 3D vision experience are required.

Qualifications

• Strong coding in Python and PyTorch.
• Video generation experience end-to-end.
• Track record of large-scale training on cluster systems.
• Working knowledge of 3D vision, multi-view geometry, scene reconstruction.

Responsibilities

• Train video and dynamics models with action conditioning for manipulation policies.
• Push long-horizon coherence by developing architectures and training methods for hard tasks.
• Own training infrastructure: run multi-GPU clusters, write custom CUDA, debug at scale.
• Build a world-model data engine to enable learning across customers and manipulation tasks.

Skills

Python

PyTorch

Video generation experience

Large-scale training

3D vision

Tools

CUDA

Job description

One Robot (YC W26) seeks an engineer to build world models that simulate manipulation scenes and validate policies in a scalable way. You will develop generative models with controllability and physical fidelity matching real-robot behavior.

You will train video and dynamics models on multi-GPU clusters, own data pipelines, and ensure learning across customers and tasks. Strong Python/PyTorch skills and video generation and 3D vision experience are required.