Founding Machine Learning - World Models

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

Compensation Range

$150K - $275K