Downloads · 30 days
12
31% of all-time downloads
twanghcmut/aam-gr00t-n16d
aam-gr00t-n16d is a robotics model from twanghcmut. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for gr00t. The card lists the license as apache-2.0.
This checkpoint adds an Action-Aware Adaptive Memory (AAM) module on top of the HAMLET window architecture for GR00T N1.6, trained on RoboMME. The memory module uses a log-spaced 8-frame window [-1024, -512, -256, -12…
Downloads · 30 days
12
31% of all-time downloads
All-time downloads
39
Public
Parameters
3.4B
10.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors10.4 GB · 100%
How the weights are stored.
F321.8B · 51%
From the Hugging Face model README
This checkpoint adds an Action-Aware Adaptive Memory (AAM) module on top of the HAMLET window architecture for GR00T N1.6, trained on RoboMME. The memory module uses a log-spaced 8-frame window [-1024, -512, -256, -128, -64, -32, -16, 0] env-steps; each past observation slot receives a learned per-slot soft gate weight, feeding a 2-layer MemoryTransformer that conditions the DiT flow-matching action head. A second prediction head performs action-conditioned forward prediction as an auxiliary loss. Trainable components: DiT + projector + moment tokens + top-4 LLM layers + gate + prediction head; the remainder of the VLM backbone is frozen.
Eval harness: 10 episodes/task, checkpoint-60000. Vanilla GR00T N1.6 = 10.12 avg, HAMLET baseline = 16.5 avg.
| Task Category | AAM (this ckpt) | HAMLET baseline |
|---|---|---|
| Counting | 5.0 | — |
| Permanence | 20.0 | ~19.5 |
| Reference | 12.5 | — |
| Imitation | 5.0 | — |
| Total avg | 10.6 | 16.5 |
The adaptive gate is strong at selective recall (Permanence ties HAMLET) but weak at temporal integration (Counting).
nvidia/GR00T-N1.6-3B