ashby
Founding Machine Learning - World Models
One Robot
San Francisco, USA
Full-time
Senior
Onsite
USD 150000-275000 yearly / year
Discovered 1 weeks ago
PythonPyTorchCUDA
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
PythonPyTorchCUDA
Key Skills for This Role
PythonPyTorchCUDA
Full Job Posting
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.
About One Robot
Robotics5 employees
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer