X-Square-Robot/wall-x

Building General-Purpose Robots Based on Embodied Foundation Model

What it solves

Wall-X provides a framework for building general-purpose robots using embodied foundation models. It aims to bridge the gap between vision-language models (VLMs) and physical interaction, allowing robots to understand how to act effectively within the physical world based on high-fidelity interaction data.

How it works

The project implements a Vision-Language-Action (VLA) architecture. It leverages large-scale multimodal pretraining to associate language and vision with specific robotic actions. The system includes specialized training and inference stacks, including flow-matching and FAST action branches, and utilizes CUDA operators for performance. It integrates with the LeRobot dataset format for data preparation and supports fine-tuning via FSDP (Fully Sharded Data Parallel) training.

Who it’s for

This is designed for robotics researchers and developers working on embodied AI, specifically those looking to deploy and fine-tune Vision-Language-Action models for real-robot manipulation and simulator evaluation.

Highlights

  • Deployment-Ready VLA: Includes the Wall-OSS-0.5 model for direct real-robot manipulation.
  • World Action Modeling: Features WALL-WM, which couples future-video imagination with action prediction.
  • Integrated Ecosystem: Provides tools for LeRobot data preparation, WebSocket serving for real-time action inference, and LIBERO simulator evaluation.
  • Performance Optimized: Uses FlashAttention and custom CUDA operators to ensure efficient inference and training.

Related

  • Project
  • Project
  • Project
  • Project
  • Project