Staff Engineer, Generative Robotics & Policy

Baseline Robotics; Zürich; Vor Ort; CHF 120.000 - 170.000; Vollzeit

Benefits dieser Stelle

Stock options

Zusammenfassung

Baseline Robotics in Zurich, Switzerland, is seeking an experienced engineer to advance diffusion-based policies on real hardware. You will focus on generative modelling for sequential decisions and publish impactful results.

You will work with Python, PyTorch, and GPU-accelerated simulators such as Isaac Lab, Newton, mjlab, Genesis, with opportunities to publish in top venues. This role is on-site in Zurich with stock options included.

Qualifikationen

• 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).

Aufgaben

• 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.

Kenntnisse

Python & PyTorch

Generative modelling

Training on real hardware

Publications

JAX or Rust

GPU-accelerated simul tools

Tools

Isaac Lab

Newton

mjlab

Genesis

Jobbeschreibung

Baseline Robotics in Zurich, Switzerland, is seeking an experienced engineer to advance diffusion-based policies on real hardware. You will focus on generative modelling for sequential decisions and publish impactful results.

You will work with Python, PyTorch, and GPU-accelerated simulators such as Isaac Lab, Newton, mjlab, Genesis, with opportunities to publish in top venues. This role is on-site in Zurich with stock options included.