🤖 nanoVLA: The Multi-Variant 3-Camera Bimanual YAM Policy Suite

Welcome to the official model repository for nanoVLA, a high-frequency, lightweight Vision-Language-Action (VLA) suite optimized for bimanual manipulation on the I2RT Bimanual YAM (Dual 7-DoF Arm) platform with 3 cameras (Overhead, Left Wrist, Right Wrist).

---

🏆 Comprehensive Architecture Leaderboard

All checkpoints below are packaged and downloadable directly from this repository:

---

🦾 Multi-Camera Coordinate Setup

                                  Prompt: "bimanual pick and place red cube"
                                                   │
    [Top Camera]         [Left Wrist Cam]    [Right Wrist Cam]     [SigLIP Text Tokens]
   (3, 224, 224)          (3, 224, 224)       (3, 224, 224)             (1, 16)
         │                      │                   │                      │
         ▼                      ▼                   ▼                      ▼
  ┌──────────────┐       ┌──────────────┐    ┌──────────────┐       ┌──────────────┐
  │ SigLIP-Base  │       │ SigLIP-Base  │    │ SigLIP-Base  │       │ Language     │
  │ Vision (86M) │       │ Vision (86M) │    │ Vision (86M) │       │ Embedding    │
  └──────┬───────┘       └──────┬───────┘    └──────┬───────┘       └──────┬───────┘
         │ 196 tokens           │ 196 tokens        │ 196 tokens           │ 16 tokens
         └──────────────────────┴──────────┬────────┴──────────────────────┘
                                           │  588 + 16 = 604 tokens
                                           ▼
                            ┌──────────────────────────────┐
                            │  nanoVLA Transformer Trunk   │
                            │  (6 Layers, 384 Dim, 6 Heads)│
                            └──────────────┬───────────────┘
                                           │
                                           ▼
                                 ┌────────────────────┐
                                 │ 14-DoF Action Head │
                                 │  Left (7) + Right(7)
                                 └────────────────────┘

---

🚀 Quickstart Inference

import torch
from model import NanoVLA, NanoVLAConfig

# Load Flagship Champion Checkpoint
ckpt = torch.load("spatial_chunked_flagship.pt", map_location="cuda")
model = NanoVLA(ckpt['config']).cuda()
model.load_state_dict(ckpt['model_state_dict'])
model.eval()

# Predict 14-DoF action for 3 cameras
imgs = torch.rand(1, 3, 3, 224, 224, device="cuda")
txt = torch.zeros(1, 16, dtype=torch.long, device="cuda")
proprio = torch.zeros(1, 14, device="cuda")

with torch.no_grad():
    actions = model.predict_action(imgs, txt, proprio_state=proprio)
print("Predicted 14-DoF Actions (7 Left + 7 Right):", actions.cpu().numpy())

---

📄 Citation & License

MIT License (c) 2026 npow. Developed for fast, accessible, and robust robot learning.