2toinf/X-VLA
[ICLR 2026] The offical Implementation of "Soft-Prompted Transformer as Scalable Cross-Embodiment Vision-Language-Action Model"
What it solves
X-VLA addresses the challenge of training generalist Vision-Language-Action (VLA) models that can work across different robotic platforms (embodiments). It leverages heterogeneous robot datasets to create a model that generalizes well across diverse hardware, improving dexterity and adaptability in robotic manipulation tasks.
How it works
X-VLA uses a soft-prompted Transformer architecture. It introduces embodiment-specific learnable embeddings (soft prompts) that guide a unified Transformer backbone to learn policies across multiple domains. To maintain consistency, it employs a unified End-Effector 6D (EE6D) control space for both proprioceptive input and action output.
Who it’s for
This project is for robotics researchers and developers who want to deploy scalable, cross-platform VLA models on physical robots or in simulation platforms like LeRobot.
Highlights
- Cross-Embodiment Generalization: Achieves state-of-the-art performance across six simulation platforms and three real-world robots.
- Server-Client Architecture: Separates the model environment from robot-specific dependencies to avoid package conflicts and support distributed inference.
- Unified Control Space: Uses a standardized EE6D control space to ensure consistency across different robotic arms.
- Extensible Training: Supports LoRA fine-tuning and a modular dataset interface for adding custom robot demonstrations.
Related
- Project
- Dispatch
- Project
- Project
- Project