alpamayo-recipes

NVlabsApache-2.0活跃维护

NVIDIA Alpamayo 的开发者配方库,提供自动驾驶与机器人模型的微调、强化学习后训练、量化和部署现成方案,省去从零搭建流程的麻烦。

Developer Hub for NVIDIA Alpamayo, containing ready-to-use recipes for fine-tuning, reinforcement-learning post-training, quantization, and deployment.

暂无版本224star +3 / 7 天54fork同步于 2026-11-11 05:59 UTC

怎么上手

由 README 提炼,执行前请确认
git clone https://github.com/NVlabs/alpamayo-recipes
cd recipes/alpamayo1_sft
硬件要求按 README 说明

README 未说明具体显存要求,各配方目录的 README 中才列出硬件要求;模型规模为 10B(Alpamayo 1/1.5 Nano)与 34B(Alpamayo 2 Super),量化与蒸馏目标为部署到 DRIVE AGX Thor。

下载:源码 zip·GitHub Releases(安装包)本站不镜像安装包

项目说明

由 README 自动翻译提炼

NVIDIA Alpamayo 后训练配方库,提供 VLA 驾驶模型的微调、强化学习、量化与蒸馏脚本

NVIDIA Alpamayo 的开发者配方库,提供自动驾驶与机器人模型的微调、强化学习后训练、量化和部署现成方案,省去从零搭建流程的麻烦。

核心能力

  • 用 Hugging Face Trainer 与 DeepSpeed 做 Alpamayo 1/1.5 监督微调
  • 用 Cosmos-RL 与 GRPO 做开环强化学习后训练
  • 用 Model Optimizer Toolkit 做 FP8 与 NVFP4 混合精度量化
  • 用 SpecForge 训练 Domino/DFlash2 投机解码草稿模型
  • 提供数据集下载、样本筛选与 checkpoint 格式转换脚本

容易踩的坑

本仓库只有配方和脚本,模型权重需从 Hugging Face 的 nvidia 组织单独下载;训练数据需先用 scripts/download_pai.py 下载 Physical AI AV 数据集;各配方的安装与硬件要求分散在子目录 README 中,需逐个查看

内容由模型根据 README 生成,可能不完整或不准确,以官方文档为准。看原文 README

最近发版

0 个版本

暂无 release 数据(可能没有发版,或尚未同步)。

用户点评

登录用户的真实使用记录,可按版本和硬件筛选
共 0 条

    还没有人点评。第一个说出真实体验的人最有价值。

    README 原文

    未翻译

    NVIDIA Alpamayo Developer Hub

    Open platform for reasoning-based autonomous driving. Vision Language Action models, closed-loop simulation, reinforcement learning, reasoning-based auto-labeling, and open driving datasets for transparent driving.

    License: Apache 2.0 arXiv Hugging Face NVIDIA Forum Contributions Welcome Challenges


    What's New

    • NVIDIA Alpamayo 2 Super is an open 34B reasoning Vision Language Action model for end-to-end autonomous driving, built on the NVIDIA Cosmos 3 backbone. Model weights and code can be found on Hugging Face and GitHub.
    • NVIDIA AlpaGym is a closed-loop reinforcement learning framework for training end-to-end autonomous driving policies in simulation.
    • NVIDIA CoC Auto-Labeling Pipeline is an open-source pipeline that generates Chain-of-Causation (CoC) reasoning labels for driving clips automatically.

    What is Alpamayo?

    NVIDIA Alpamayo is an open platform designed to accelerate the development of safe, transparent, and reasoning-based autonomous vehicles. It consists of vision-language-action (VLA) models, simulation frameworks, reinforcement learning infrastructure, and physical AI datasets and tools. The VLA models scale from the 10B-parameter Alpamayo 1 Nano and Alpamayo 1.5 Nano to the 34B-parameter Alpamayo 2 Super, each generating driving trajectories alongside CoC reasoning traces that make every decision transparent and auditable. AlpaSim provides closed-loop policy simulation, and AlpaGym enables reinforcement learning on top to learn from edge-case failures across massively parallel GPU simulations. The Physical AI AV Dataset spans 2500+ cities across 25 countries and provides real-world long-tail training data. The CoC Auto-Labeling Pipeline annotates data with meta-actions and reasoning labels automatically.


    Why Alpamayo?

    Open platformModel weights, inference code, closed-loop simulation, reinforcement learning tools, and driving datasets, all published openly for reasoning-based autonomous driving
    Reasoning-basedCoC traces accompany every predicted trajectory, making each driving decision transparent and auditable
    Closed-loop by designAlpaGym and AlpaSim train and validate driving policies through continuous decision and observation cycles, surfacing the compounding, long-tail failures that open-loop log replay and imitation learning miss
    Scales with GPUsSimulation, reinforcement learning, and inference scale from a single GPU to multi-node across the family

    Which Repository should I use?

    This repository holds the post-training recipes and utility scripts. The models, simulation frameworks, and datasets live in the sibling repositories linked in each section below.

    PurposeWhen to useLocation
    InferenceRun inference and reasoning with the Alpamayo VLA modelsYou want to run or evaluate a version of an Alpamayo VLA modelNVlabs/alpamayo, NVlabs/alpamayo1.5, NVlabs/alpamayo2
    RecipesPost-train, quantize or distill Alpamayo VLA modelsYou want to adapt an Alpamayo model to your data, compute budget, or sensor rigThis repository
    Simulation and RLTest and train driving policies in closed-loop simulationYou want to validate policies or run RL beyond open-loop log replayNVlabs/alpasim, NVlabs/alpagym
    Data and LabelingGenerate reasoning labelsYou want to auto-label driving clips or start from open dataNVlabs/alpamayo-coc-autolabeler
    Physical AI AV DatasetOpen multi-sensor driving dataset with CoC labelsYou want real-world training data to train or test your modelsNVlabs/physical_ai_av

    Recipes

    End-to-end recipes for the Alpamayo VLA models, covering supervised fine-tuning (SFT), open-loop reinforcement learning (RL), and quantization. Each recipe includes:

    • Data preparation using the utility scripts to download and curate the Physical AI Autonomous Vehicles dataset
    • Training configuration with model, optimizer, and hyperparameter settings
    • A per-folder README with installation, run instructions, and hardware requirements
    RecipeDescriptionStage
    recipes/alpamayo1_sft/Alpamayo 1 supervised fine-tuning with Hugging Face Trainer and DeepSpeedSFT
    recipes/alpamayo1_5_sft/Alpamayo 1.5 supervised fine-tuning with Hugging Face Trainer and DeepSpeedSFT
    recipes/alpamayo1_x_rl/Alpamayo 1 and 1.5 open-loop reinforcement learning post-training with Cosmos-RL and GRPORL (GRPO)
    recipes/alpamayo_distillation/Alpamayo 1 and 1.5 distillation, with optional expert SFT and RL refinement
    recipes/alpamayo1_5_speculative_decoding/Alpamayo 1.5 Domino and DFlash2 draft training with SpecForgeSpeculative Decoding
    recipes/alpamayo1_5_quant/Alpamayo 1.5 quantization with Model Optimizer Toolkit — FP8 and NVFP4 + FP8 Mixed PrecisionQuantization

    What you can adapt: fine-tune on your own fleet data, adjust the camera count and sensor configuration, and swap reward functions for the RL recipe. See each recipe README for supported inputs and hardware requirements.


    Utility Scripts

    Helper scripts for preparing data and converting checkpoints across the Alpamayo recipes.

    ScriptPurpose
    scripts/download_pai.pyDownload the Physical AI Autonomous Vehicles dataset from Hugging Face
    scripts/curate_pai_samples.pyCurate a subset of Physical AI Autonomous Vehicles samples
    scripts/convert_checkpoint.pyConvert between Alpamayo 1 and 1.5 checkpoints
    scripts/convert_release_config_to_training.pyConvert a release checkpoint to training format
    scripts/convert_cosmos_rl_checkpoint.pyConvert a Cosmos-RL checkpoint to Hugging Face format

    Alpamayo VLA Models

    Vision Language Action models that generate driving trajectories alongside CoC reasoning traces for end-to-end autonomous driving.

    ModelParamsBackboneHighlightsResources
    Alpamayo 1 Nano10BCosmos-ReasonFour-camera reasoning VLA with trajectory and reasoning outputNVlabs/alpamayo, nvidia/Alpamayo-R1-10B
    Alpamayo 1.5 Nano10BCosmos-Reason2RL post-trained, flexible camera count, navigation guidance, visual question answeringNVlabs/alpamayo1.5 ,nvidia/Alpamayo-1.5-10B
    Alpamayo 2 Super34BCosmos 3360-degree surround-view camera inputs, meta-action outputs, CoC autolabeling, visual question answering, 2D grounding, multi-task teacher modelNVlabs/alpamayo2 , nvidia/Alpamayo2-Super

    White paper: Alpamayo-R1: Bridging Reasoning and Action Prediction for Generalizable Autonomous Driving in the Long Tail

    NVIDIA Alpamayo 2 Super is an open 34B-parameter reasoning Vision Language Action (VLA) model for end-to-end autonomous driving, built on the NVIDIA Cosmos 3 backbone at 3x the scale of prior Alpamayo generations. Full 360-degree surround perception and meta-action outputs encoding high-level behavioral decisions alongside trajectory predictions make it multi-task by design, handling planning, CoC auto-labeling, grounded visual question answering, and model evaluation in one architecture.

    Model Specifications:

    • 34B parameters, built on the Cosmos 3 backbone, 3x the scale of Alpamayo 1 Nano and 1.5 Nano
    • 360-degree surround perception
    • Meta-action outputs encoding high-level behavioral decisions alongside trajectory predictions
    • Trained on 110,000+ hours of driving data
    • Multi-task: planning, CoC auto-labeling, and grounded visual Q&A in one architecture

    What You Can Build:

    • Trajectory planning with explicit meta-actions
    • CoC auto-labeling from driving clips
    • Evaluation and judging of smaller on-board models
    • A teacher model for distillation and quantization into student models that meet in-vehicle latency and safety requirements on DRIVE AGX Thor

    Resources: NVlabs/alpamayo2 · nvidia/Alpamayo2-Super

    NVIDIA Alpamayo 1.5 Nano is an open 10B-parameter reasoning Vision Language Action (VLA) model for end-to-end autonomous driving, reinforcement-learning post-trained on the NVIDIA Cosmos-Reason2 backbone. RL post-training sharpens reasoning quality and tightens consistency between what the model decides and what it does. Flexible multi-camera support with variable camera count adapts the model to different sensor rigs without retraining from scratch. Visual question answering and natural-language navigation make the vehicle steerable and explainable in real time.

    Model Specifications:

    • 10B parameters, built on the Cosmos-Reason2 backbone
    • RL post-trained for reasoning quality and reasoning-trajectory consistency
    • Flexible multi-camera support with variable camera count
    • Visual question answering and natural-language navigation guidance
    • SFT and RL post-training scripts included

    Resources: NVlabs/alpamayo1.5 · nvidia/Alpamayo-1.5-10B

    NVIDIA Alpamayo 1 Nano is an open 10B-parameter reasoning Vision Language Action (VLA) model for end-to-end autonomous driving, built on the NVIDIA Cosmos-Reason backbone. It was the first open industry-scale reasoning VLA for autonomous driving, released as Alpamayo-R1 and introduced as Alpamayo 1 at CES 2026. It generates driving trajectories alongside CoC reasoning traces from multi-camera video and egomotion history.

    Model Specifications:

    • 10B parameters, Cosmos-Reason backbone (8.2B) with a diffusion-based action expert (2.3B)
    • Inputs: four-camera video, egomotion history, and text commands
    • Outputs: driving trajectories and natural-language CoC reasoning traces
    • Trained on more than 1B images from 80,000 hours of driving
    • Runs on a single GPU with 24GB of memory

    Resources: NVlabs/alpamayo · nvidia/Alpamayo-R1-10B


    Simulation and Reinforcement Learning

    Closed-loop simulation and reinforcement learning for autonomous driving policies, where every policy decision reshapes the simulation state.

    FrameworkRoleHighlightsResources
    AlpaSimClosed-loop simulation platformMicroservice architecture, NuRec and OmniDreams rendering backends, GPU-scalableNVlabs/alpasim
    AlpaGymClosed-loop RL frameworkTrains driving policies in AlpaSim, Cosmos RL backend, GRPO default, swappable rewards and algorithmsNVlabs/alpagym

    NVIDIA AlpaSim is an open-source closed-loop simulation platform for developing and testing end-to-end autonomous driving policies. Its microservice architecture orchestrates Driver, Renderer, TrafficSim, Controller, and Physics services as independent processes, each assignable to different GPUs for parallelized simulation across large fleets of concurrent policy runs.

    What It Provides:

    • Closed-loop simulation where every policy decision reshapes the simulation state, going beyond open-loop log replay
    • GPU-scalable microservice architecture across independent services
    • Two rendering backends: Omniverse NuRec for neural reconstruction of real-world scenes, and OmniDreams for generative world model rendering of novel and long-tail scenarios
    • The orchestration layer for AlpaGym reinforcement learning, scaling from single-GPU development to multi-node training

    Resources: NVlabs/alpasim

    NVIDIA AlpaGym is an open-source closed-loop reinforcement learning (RL) framework for training end-to-end autonomous driving policies at GPU scale. It trains policies through continuous closed-loop decision and observation cycles inside AlpaSim, where every braking, steering, and navigation action reshapes the environment and surfaces the compounding, long-tail failures that log-based imitation learning misses.

    What It Provides:

    • Closed-loop reinforcement learning across parallel GPU simulations
    • Built on AlpaSim with NVIDIA Cosmos RL as the backend and GRPO as the default RL algorithm
    • Swappable policy models, reward configurations, and RL algorithms
    • Standard reward functions for collision avoidance, offroad detection, and progress
    • Physical AI NuRec dataset support to begin RL training without additional scene reconstruction or annotation

    Resources: NVlabs/alpagym


    Data, Labels, and Benchmarks

    Open Datasets

    Open multi-sensor autonomous driving datasets and synthetic scenarios for training and validating reasoning-based driving systems.

    DatasetUsageLicenseDescription
    PhysicalAI-Autonomous-VehiclesTrainingNVIDIA AV Dataset License1,700+ hours of multi-sensor driving data across 25 countries, 306K clips, with 7 synchronized cameras, LiDAR, and radar
    PhysicalAI-Autonomous-Vehicles-NuRecSimulation / RLNVIDIA AV Dataset LicenseNeural reconstruction scenarios paired to AV clips, the default AlpaGym RL starting dataset, with ClipGT annotations
    PhysicalAI-Autonomous-Vehicle-Cosmos-Drive-DreamsSDGCC-BY-4.081K synthetic videos with LiDAR and HD-map annotations
    PhysicalAI-Autonomous-Vehicle-Cosmos-SyntheticSDGCC-BY-4.0Cosmos-generated synthetic driving scenarios

    CoC reasoning labels are being added to the Physical AI Open Dataset on Hugging Face. See the Physical AI Autonomous Vehicles collection.

    Reasoning Labels

    Automated reasoning-label generation and open datasets for reasoning-based autonomous driving.

    ToolRoleHighlightsResources
    CoC Auto-Labeling PipelineCoC label generationThree-step keyframe-to-trace pipeline, VLM-based, causally grounded reasoning labelsNVlabs/alpamayo-coc-autolabeler

    NVIDIA CoC Auto-Labeling Pipeline is an open-source auto-labeling pipeline that generates CoC reasoning labels for driving clips automatically. It identifies decision-making keyframes from high-level motion data, runs a VLM pipeline on those keyframes, and organizes cause and effect into decision-grounded, causally linked traces. The output is a strong foundation for training reasoning-based driving models such as Alpamayo 2 Super.

    Three-Step Process:

    1. Identify the decision-making moment. Keyframes are defined by critical objects, traffic lights, yield and stop signs, road events, lane lines, and ODD events, detected through sudden changes in egomotion.
    2. Label the explicit driving decision. Longitudinal and lateral decisions are drawn from a closed action set that eliminates vague behavior descriptions.
    3. Organize cause and effect into a CoC trace. Causal factors are linked to the resulting decision.

    Highlights:

    • Automated CoC labeling at scale
    • Causally grounded traces that reflect the actual reason for each decision
    • Closed action set for precise, unambiguous driving decisions
    • Used to generate CoC reasoning labels in the NVIDIA Physical AI Open Dataset

    Resources: NVlabs/alpamayo-coc-autolabeler

    🏆 Participate in Challenges

    Test your models against the Alpamayo ecosystem benchmarks — open to the community.

    ChallengeDescription
    Physical AI AV Out-of-Distribution Reasoning Challenge 2026Evaluate VLA model reasoning on rare and long-tail driving scenarios from the Physical AI dataset
    AlpaSim End-to-End Closed-Loop Challenge 2026Benchmark end-to-end driving policies in closed-loop simulation with AlpaSim

    Technical Blogs


    Contributing

    Contributions are welcome, including examples, recipes, and tooling. Please read the Contributing Guidelines before submitting pull requests.


    Support

    📣 Usage questions and discussion: join us on the Alpamayo NV Developer Forum.

    🐛 Bugs, documentation issues, and feature requests: file a GitHub issue using the appropriate template. The relevant NVIDIA responder is auto-assigned.


    Security

    To report a vulnerability, please contact [email protected] or use NVIDIA's Vulnerability Disclosure Program. Do not file security issues publicly.


    License

    Code in this repository is released under the Apache 2.0 License. See LICENSE for details. Model weights and datasets are governed by their respective licenses listed above.


    NVIDIA Alpamayo. Open family for reasoning-based autonomous driving.