alpamayo1-x-rl

Official

Automate RL post-training for Alpamayo models

AuthorNVlabs
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill automates the process of running end-to-end open-loop RL post-training for Alpamayo VLM action models on the Physical AI AV (PAI) dataset using Cosmos-RL + GRPO.

Core Features & Use Cases

  • Convert release checkpoint to training-ready format: Converts Alpamayo model checkpoints to a training-ready format.
  • Curate PAI mini subset: Extracts a small subset of the PAI dataset for local training.
  • Launch RL training: Runs RL training on the dataset using Cosmos-RL and GRPO.
  • Export trained policy checkpoint: Converts the trained policy checkpoint to a HuggingFace checkpoint directory for inference.

Quick Start

Use the alpamayo1-x-rl skill to train an Alpamayo 1.5 model using the PAI dataset and export the resulting policy checkpoint to a HuggingFace directory.

Dependency Matrix

Required Modules

alpamayo_r1alpamayo-recipescosmos-rlflash-attn

Components

scriptsreferencesassets

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: alpamayo1-x-rl
Download link: https://github.com/NVlabs/alpamayo-recipes/archive/main.zip#alpamayo1-x-rl

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 620,000+ vetted skills library on demand.