sii-research/tau-0-vla

This repo is the official implementation of "τ0-VLA: a Hierarchical Robot Foundation Model with World-Model-Guided Test-Time Computation".

What it solves

τ₀-VLA is designed for long-horizon robot manipulation tasks. It addresses the challenge of executing complex sequences of actions across different robot embodiments by combining high-level reasoning and low-level execution.

How it works

The model uses a hierarchical architecture:

  1. High-Level Policy: A memory-augmented policy that determines the next subtask. It employs world-model-guided test-time computation to search for the best alternative actions when complex reasoning is required.
  2. Low-Level Policy: A generalist policy that executes the chosen subtask. This policy is built on a Qwen3.5 vision-language backbone and uses a Mixture-of-Transformers action expert trained via conditional flow matching. It operates on a unified 40-dimensional state/action space and was trained on over 40,000 hours of diverse real-world robot data.

Who it’s for

This project is for robotics researchers and developers working on foundation models for robot control, long-horizon manipulation, and generalist robot policies.

Highlights

  • Hierarchical Structure: Separates high-level subtask planning from low-level execution.
  • World-Model Guidance: Uses test-time computation to refine subtask selection through search.
  • Coflow Matching: Employs conditional flow matching for the action expert.
  • Cross-Embodiment: Trained on heterogeneous real-world data to work across different robot types.

Related

  • Project
  • Project
  • Project
  • Project
  • Project