MA-WAM checkpoints

This repository contains 30 world-model checkpoints across MPE, SMAC and MAMuJoCo. The archive preserves the original PyTorch checkpoint files. MANIFEST.json records each file's path, size and SHA256. Extract MA-WAM-weights.zip before use.

Scope

This is the primary component checkpoint bundle. It does not include the full collection of ablation models or every policy training seed. Use the matching architecture and environment dimensions from the project code when loading checkpoints.

CoFlow backbone and datasets

Use CoFlow-checkpoints for the shared policy backbone where the required variant, task, quality, seed and step match. Policy weights are not duplicated here. Matching checkpoint steps alone do not establish identical weights.

Datasets are not uploaded here. Refer to CoFlow-datasets and the CoFlow data instructions.

Links

• Paper (arXiv:2609.31281)

• Project page

• Code

• Maintainer: Guowei Zou, Sun Yat-sen University