Skip to page content
SAVRN
Search Plan a project

Open-weight model · Robotics

so101-stack3-pi05-baseline-4k

by Mai Jiahong WoWjx/so101-stack3-pi05-baseline-4k

so101-stack3-pi05-baseline-4k is an open-weight model for robotics from Mai Jiahong. It has 4.1B parameters. At 16-bit it needs about 9.9 GB of GPU memory, which fits on 1x MI300X from $1.85 an hour, at the lowest prices in the SAVRN Index.

Inference checkpoint from an early Stack3 fine-tune of a two-camera SO-101 π0.5 policy.

Parameters4.1B
Context—
Weights18.7 GB
License—
AccessOpen weights
Monthly Downloads—

Runs On

What it takes to serve so101-stack3-pi05-baseline-4k (4.1B parameters): the memory its weights need at each precision, and the cheapest way to rent enough data-center GPUs to hold them.

PrecisionWeightsMemory neededCheapest setupPer hourAlso fits
16-bit 8.3 GB 9.9 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
8-bit 4.1 GB 5.0 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00
4-bit 2.1 GB 2.5 GB 1x MI300X (192 GB)
Vultr
$1.85 1x H100 $1.99 · 1x MI325X $2.00

Memory is the weights at that precision plus 20% for the runtime and a short context; a long context needs more. Prices are the lowest on-demand hourly rates in the SAVRN Index, read Oct 18, 2026.

so101-stack3-pi05-baseline-4k on every accelerator the SAVRN Index prices, at every precision

Model Card

Inference checkpoint from an early Stack3 fine-tune of a two-camera SO-101 π0.5 policy. It is warm-started from the Stack2 checkpoint that performed well in the user's real-robot trials, followed by 4,000 optimizer updates on the collected Stack3 full-chain dataset. There is no held-out validation set in this initial run. Training health or loss does not establish successful three-block stacking. This checkpoint has not been verified on the physical robot; it is provided for deployment testing, not as a claim of task success. The repository contains policy weights, model configuration, tokenizer, and matching observation/action pre- and post-processors with the normalization statistics…

Excerpt from the card by Mai Jiahong.

Identity and Version

Repository
WoWjx/so101-stack3-pi05-baseline-4k
Publisher
Mai Jiahong
Task
Robotics
Modality
Control
Library
lerobot
Parameters
4.1B parameters
Languages
Not stated by the source
Revision
cdaf709fa64744e2cf6b6389013bd23b5383c93c
First published
2026-10-05
Last updated
2026-10-06

Files and Weights

21 files, 18.8 GB in total. The weights are 6 files totalling 18.7 GB in safetensors.

Weights6 files · 18.7 GB
Configuration8 files · 15.0 KB
Tokenizer4 files · 68.8 MB
Documentation2 files · 2.3 KB
Repository1 file · 1.7 KB
Every file
FileTypeSizeSHA-256
checkpoints/002000/model.safetensorsWeights9.4 GB 02fed500fc4f
checkpoints/002000/policy_postprocessor_step_0_unnormalizer_processor.safetensorsWeights7.6 KB cefdde799aef
checkpoints/002000/policy_preprocessor_step_3_normalizer_processor.safetensorsWeights7.5 KB e16f77f85ac2
model.safetensorsWeights9.4 GB b47abef150bf
policy_postprocessor_step_0_unnormalizer_processor.safetensorsWeights7.6 KB cefdde799aef
policy_preprocessor_step_3_normalizer_processor.safetensorsWeights7.5 KB e16f77f85ac2
SHA256SUMS.jsonConfiguration1.4 KB —
checkpoints/002000/config.jsonConfiguration2.8 KB —
checkpoints/002000/policy_postprocessor.jsonConfiguration778 B —
checkpoints/002000/policy_preprocessor.jsonConfiguration2.2 KB —
checkpoints/006000/config.jsonConfiguration2.7 KB —
config.jsonConfiguration2.3 KB —
policy_postprocessor.jsonConfiguration778 B —
policy_preprocessor.jsonConfiguration2.2 KB —
README.mdDocumentation1.9 KB —
checkpoints/002000/README.mdDocumentation406 B —
.gitattributesRepository1.7 KB —
checkpoints/002000/tokenizer/tokenizer.jsonTokenizer34.4 MB 7336bb4ce016
checkpoints/002000/tokenizer/tokenizer_config.jsonTokenizer502 B —
tokenizer/tokenizer.jsonTokenizer34.4 MB 7336bb4ce016
tokenizer/tokenizer_config.jsonTokenizer693 B —

License and Download

License
Not stated by the source
Access
Open weights, no gate
Download size
18.7 GB
Download from Mai Jiahong

Released by Mai Jiahong through its official repository on Hugging Face.

Memory Requirements

PrecisionWeights in memory
As published18.7 GB
16-bit8.3 GB
8-bit4.1 GB
4-bit2.1 GB

Weights only, from the published parameter count; the key-value cache and runtime add to this.

Questions About so101-stack3-pi05-baseline-4k

How much GPU memory does so101-stack3-pi05-baseline-4k need?

About 9.9 GB at 16-bit and 2.5 GB at 4-bit: the weights (4.1B parameters) plus a working margin. A long context needs more.

What is the cheapest GPU to run so101-stack3-pi05-baseline-4k on?

At 16-bit, 1x MI300X from $1.85 an hour; at 4-bit, 1x MI300X from $1.85 an hour, at the lowest on-demand prices the SAVRN Index lists.

Similar Models

Model · Robotics

unlv_vla_policy

Jinseok Kim

π₀.₅ (Pi05) is a Vision-Language-Action model from Physical Intelligence designed for open-world generalization: it evolves π₀ to generalize to entirely new environments and situations that were never seen during training. The LeRobot implementation is adapted from their open-source OpenPI repository. This policy has been trained and pushed to the Hub using LeRobot. Learn how to train and run it in the LeRobot pi05 guide, or browse the full documentation. The policy consumes these observation features and produces these action features. Inputs Outputs New to LeRobot? These guides cover the full workflow: - Install LeRobot — set up the lerobot package. - Hardware setup — assemble, wire, and…

Open weights apache-2.0 4.1B parameters lerobot

π₀.₅ (lerobot/pi05base) full fine-tune on mliu48/telearms-yam-cube-v4 (53,832 bimanual YAM demonstrations, 48 tasks, 30 Hz, 3 × 224×224 cameras). LeRobot 0.6.1 @ e40b58a8 with the repo's patch (tools/installlerobotpatch.py). - Full fine-tune (LLM + vision + action expert), bf16, gradient checkpointing. - 16 × H100 (2 nodes × 8), batch 16/GPU (global 256), 99,486 steps (one selected-data epoch). - Peak LR 7.07e-5, cosine decay to 7.07e-06, 1,000 warmup steps; relative actions (absolute grippers). Final checkpoint (step 99,486), training/evaluatepi.py suite settings in the RoboEval YAM simulator: 10 episodes per task, seeds 900000001–900000010, 1,800-step limit, 25-step action chunks, success…

Open weights 4.1B parameters lerobot

Final checkpoint after 60,000 optimization steps. This is a trained policy, not an evaluation result. Policy weights, policy configuration, preprocessing/postprocessing and normalization states are at the repository root. Training resume files are in trainingstate/. Host-specific paths were removed from JSON metadata; supply local dataset/output paths when resuming. The tokenizer reference points to google/paligemma-3b-pt-224 (training revision 35e4f46485b4d07967e7e9935bc3786aad50687c). For subtask models, supply the corresponding per-frame subtask as the policy task text. For 3-view models also supply the dataset-defined keyframe image. No real-robot evaluation metrics are claimed here.…

Open weights 4.1B parameters lerobot

Final checkpoint after 60,000 optimization steps. This is a trained policy, not an evaluation result. Policy weights, policy configuration, preprocessing/postprocessing and normalization states are at the repository root. Training resume files are in trainingstate/. Host-specific paths were removed from JSON metadata; supply local dataset/output paths when resuming. The tokenizer reference points to google/paligemma-3b-pt-224 (training revision 35e4f46485b4d07967e7e9935bc3786aad50687c). For subtask models, supply the corresponding per-frame subtask as the policy task text. For 3-view models also supply the dataset-defined keyframe image. No real-robot evaluation metrics are claimed here.…

Open weights 4.1B parameters lerobot

π₀.₅ (Pi05) is a Vision-Language-Action model from Physical Intelligence designed for open-world generalization: it evolves π₀ to generalize to entirely new environments and situations that were never seen during training. The LeRobot implementation is adapted from their open-source OpenPI repository. This policy has been trained and pushed to the Hub using LeRobot. Learn how to train and run it in the LeRobot pi05 guide, or browse the full documentation. The policy consumes these observation features and produces these action features. Inputs Outputs New to LeRobot? These guides cover the full workflow: - Install LeRobot — set up the lerobot package. - Hardware setup — assemble, wire, and…

Open weights apache-2.0 4.1B parameters lerobot

π₀.₅ (Pi05) is a Vision-Language-Action model from Physical Intelligence designed for open-world generalization: it evolves π₀ to generalize to entirely new environments and situations that were never seen during training. The LeRobot implementation is adapted from their open-source OpenPI repository. This policy has been trained and pushed to the Hub using LeRobot. Learn how to train and run it in the LeRobot pi05 guide, or browse the full documentation. The policy consumes these observation features and produces these action features. Inputs Outputs New to LeRobot? These guides cover the full workflow: - Install LeRobot — set up the lerobot package. - Hardware setup — assemble, wire, and…

Open weights apache-2.0 4.1B parameters lerobot