Staff Engineer - Generative Robot Policies (Zurich On-Site)

Baseline Robotics; Zürich; On-site; CHF 120,000 - 170,000; Full time

Benefits offered by this job

Stock options

Job summary

Baseline Robotics in Zurich, Switzerland, is seeking a specialist to train diffusion and flow-matching policies on demonstration data and to tune action representation, chunking, and sampling on real hardware.

You will work on out-of-distribution handling through the policy's own previous actions, with total compensation CHF 120,000–170,000 plus stock options. Applications are reviewed on a rolling basis.

Qualifications

• Experience training and evaluating generative robot policies on physical hardware.
• Publications in top ML or robotics venues, or equivalent experience.
• Fluency in Python and PyTorch.

Responsibilities

• Train diffusion and flow-matching policies on demonstration data.
• Tune action representation, chunking, and sampling on real hardware.
• Handle out-of-distribution states through the policy's own prior actions.

Skills

Python

PyTorch

Robotics policy training

Publications

Job description

Baseline Robotics in Zurich, Switzerland, is seeking a specialist to train diffusion and flow-matching policies on demonstration data and to tune action representation, chunking, and sampling on real hardware.

You will work on out-of-distribution handling through the policy's own previous actions, with total compensation CHF 120,000–170,000 plus stock options. Applications are reviewed on a rolling basis.