Member of Technical Staff, World Models

About Baseline Robotics Open source revolutionised how we write software, but its impact on robotics stays limited due to expensive hardware accessible only to well-funded labs. We run that hardware as shared infrastructure for the broader community, billed only for active compute. For a given task, policies are evaluated under similar conditions, giving researchers common ground for fair comparisons.

In this role you will

• Build action-conditioned world models predicting dynamics from camera and sensor data.
• Use model rollouts for planning and post-training reference policies.
• Generate synthetic trials from the model to expand coverage where real data is scarce.
• Diagnose where model predictions diverge from the stations and trace the cause.

What we hope you'll bring

• Experience training world models, video prediction, or latent dynamics models on real data.
• Knowledge of model-based reinforcement learning, planning, and sequence modelling.
• Publications in top machine learning or robotics venues, or equivalent experience.
• Fluency in Python and PyTorch.
• Experience with JAX or Rust.
• Experience with GPU-accelerated simulation tools (Isaac Lab, Newton, mjlab, Genesis).

We encourage you to apply even if you do not believe you meet every qualification.

Details

• Location: Zurich, Switzerland, on-site 5 days a week.
• Total compensation: CHF 120,000 - 160,000.
• Every full-time hire receives stock options as part of the stated total compensation.

Deadline to apply: None. Applications will be reviewed on a rolling basis.