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 addresses the challenge of long-horizon robot manipulation tasks. It provides a hierarchical framework that can plan high-level subtasks and execute them through a generalist low-level policy that works across different robot embodiments.

How it works

The system uses a two-tier architecture:

  1. High-Level Policy: A memory-augmented planner (Proposal) that predicts the next subtask. It uses a world model to generate goal images, allowing the system to search for better alternatives when complex reasoning is required during test-time computation.
  2. Low-Level Policy: A generalist executor that converts subtasks into robot actions. It combines a Qwen3.5 vision-language backbone with a Mixture-of-Transformers action expert trained via conditional flow matching on over 40,000 hours of diverse real-world robot data.

Who it’s for

Researchers and developers working on robot foundation models, embodied intelligence, and long-horizon manipulation tasks across various robot hardware platforms.

Highlights

  • Hierarchical Structure: Separates high-level planning from low-level execution for better long-term task management.
  • World-Model Guidance: Uses a world model to generate goal images for test-time reasoning and subtask refinement.
  • Cross-Embodiment: A unified 40-dimensional state/action space allows the low-level policy to be used across different robots.
  • Extensive Training: Pretrained on a massive dataset of 40,115 hours of heterogeneous real-world robot data.

Related

  • Project
  • Dispatch
  • Project
  • Project
  • Project