Member of Technical Staff, Generative Policies

About Baseline

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

• Train diffusion and flow-matching policies on demonstration data.
• Tune action representation and chunking on real hardware.
• Condition policies on visual history, proprioception, and language.
• Choose sampling to preserve action multimodality within the control budget.
• Recover from the off-distribution states a policy reaches by acting on prior outputs.

What we hope you'll bring

• Experience training and evaluating generative policies on real hardware.
• Knowledge of generative modelling for sequential decision problems.
• 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, on-site 5 days a week.
• Annual salary: CHF 120,000 - 170,000.
• Every full-time hire receives stock options as part of the stated salary.

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