# Founding Machine Learning - World Models

**Company:** [One Robot](https://zerogtalent.com/frontier-companies/one-robot)
**Location:** San Francisco, CA, US
**Work mode:** On-site
**Seniority:** Entry
**Employment type:** Full Time
**Salary:** USD 150K–275K/yr
**Category:** Software
**Listed:** 2026-04-30

[View and apply on Zero G Talent](https://zerogtalent.com/frontier-jobs/one-robot/founding-machine-learning-world-models-93969)

## Job description

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.

## Other open roles at One Robot

- [Machine Learning Intern](https://zerogtalent.com/frontier-jobs/one-robot/machine-learning-intern-96233) — San Francisco, CA, US
- [Founding Robot Learning](https://zerogtalent.com/frontier-jobs/one-robot/founding-robot-learning-93971) — San Francisco, CA, US
- [Founding Machine Learning - Eval Layer](https://zerogtalent.com/frontier-jobs/one-robot/founding-machine-learning-eval-layer-93970) — San Francisco, CA, US

---
*Source: [Zero G Talent](https://zerogtalent.com), the job board for space, defense, AI, robotics and frontier-tech companies.*