Research Scientist - Robot Learning (VLA / WAM)

SpAItial AI; München; On-site; EUR 120,000 - 180,000; Full time

Job summary

SpAItial AI seeks a Research Scientist to train policies that turn world models into actions. You will own VLA and WAM pipelines end to end, from data to policy execution on robots, and push the boundary of embodied AI research.

The role involves closing the sim-to-real gap, adapting VLM backbones for control, and coordinating diverse robot datasets. A PhD plus industry experience is preferred, with expert Python and PyTorch skills.

Qualifications

• PhD in robotics, machine learning or computer vision with a robot learning focus.
• Publications at top venues or deployed systems.
• Deep experience with modern robot policy designs (VLA, WAM, diffusion).
• Strong imitation learning fundamentals and RL fine-tuning of pretrained policies.
• Fluency with VLM backbones and adapting them for control.
• Expert Python and PyTorch, with multi-node distributed training experience (FSDP or equivalent).

Responsibilities

• Own the training pipeline for VLA and WAM end to end, from data to a policy running on a robot.
• Contribute to setting the technical direction for embodied research at SpAItial.
• Close the sim-to-real gap through domain randomization, system identification, and calibration, and build evaluation that predicts real-world transfer.
• Adapt VLM backbones for control: encoder choice and adapter strategies, co-training.
• Curate and weight the training mix across heterogeneous robot datasets, spanning differing embodiments, action spaces, and sensor setups.
• Design action representation and decoding, including tokenization, chunking, diffusion, and flow-matching action experts.
• Build the world-model components that predict future observations conditioned on action.
• Run post-training: supervised fine-tuning onto target embodiments, and RL for robustness beyond demonstrations.

Skills

Robot policy design

Python

PyTorch

Distributed training

Imitation learning

Reinforcement learning

VLA/WAM

VLM backbones

Diffusion models

Education

PhD in robotics, ML, or CV with robot learning focus

Tools

Distributed training tooling

Job description

SpAItial is pioneering the next generation of World Models, pushing the boundaries of generative AI, computer vision, and the simulation of reality. We are moving beyond 2D pixels to build models that natively understand the physics and geometry of our world. Our mission is to redefine how industries, from robotics and AR/VR to gaming and cinema, generate and interact with physically-grounded 3D environments.

We're seeking a Research Scientist to train the policies that turn a world model into a robot that acts. You will own vision-language-action (VLA) and world-action models (WAM) end to end, starting, including data, backbone, action representation, training runs, and the evaluation that tells us whether a policy is genuinely competent or merely lucky. A world model that understands geometry and physics still doesn't act on its own; the policy is what closes that gap. This is a senior, hands-on research role for someone who has already trained manipulation policies that worked, and who can say precisely why the ones that didn't failed.

Responsibilities

• Own the training pipeline for vision-language-action (VLA) and world-action models (WAM) end to end, from data to a policy running on a robot.
• Contribute to setting the technical direction for embodied research at SpAItial.
• Close the sim-to-real gap through domain randomization, system identification, and calibration, and build evaluation that predicts real-world transfer.
• Adapt VLM backbones for control: encoder choice and adapter strategies, co-training.
• Curate and weight the training mix across heterogeneous robot datasets, spanning differing embodiments, action spaces, and sensor setups.
• Design action representation and decoding, including tokenization, chunking, diffusion, and flow-matching action experts.
• Build the world-model components that predict future observations conditioned on action.
• Run post-training: supervised fine-tuning onto target embodiments, and RL for robustness beyond demonstrations.

Key Qualifications

• A PhD in robotics, machine learning, or computer vision with a robot learning focus, from the PhD alone or followed by industry experience.
• Publications at top venues such as (CoRL, RSS, ICRA, IROS or CVPR, ICCV, ECCV, NeurIPS), open-source work, and/or deployed systems.
• Deep experience with modern robot policy designs (VLA, WAM, diffusion), trained end to end rather than fine-tuned from a released checkpoint.
• Strong imitation learning fundamentals, and familiarity with RL fine-tuning of pretrained policies.
• Fluency with VLM backbones and how to adapt them for control.
• Expert Python and PyTorch, with multi-node distributed training experience (FSDP or equivalent).

At SpAItial, we are committed to creating a diverse and inclusive workplace. We welcome applications from people of all backgrounds, experiences, and perspectives. We are an equal opportunity employer and ensure all candidates are treated fairly throughout the recruitment process.