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