AI Jobs Map

Baseline Robotics · Zurich, Zurich, Switzerland

Member of Technical Staff, Generative Policies

directorfull timePosted 7 days ago
Apply on LinkedInLinkedInOpens the original posting. AI Jobs Map never asks for your details.

Stack mentioned

machine-learningpythonpytorchrust

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

- 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, Switzerland, on-site 5 days a week.

- Total compensation: CHF 120,000 - 170,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.

More jobs at Baseline Robotics