├── .gitignore
├── LICENSE
├── README.md
├── assets
├── demo.gif
├── img.png
└── img_2.png
├── config
├── agent
│ └── racing.yaml
├── env
│ └── racing.yaml
├── trainer.yaml
└── world_model_env
│ └── fast.yaml
├── game
└── spawn
│ ├── 1
│ ├── act.npy
│ ├── full_res.npy
│ ├── low_res.npy
│ └── next_act.npy
│ ├── 2
│ ├── act.npy
│ ├── full_res.npy
│ ├── low_res.npy
│ └── next_act.npy
│ └── 3
│ ├── act.npy
│ ├── full_res.npy
│ ├── low_res.npy
│ └── next_act.npy
├── requirements.txt
├── scripts
├── import_run.py
└── resume.sh
└── src
├── __init__.py
├── agent.py
├── coroutines
├── __init__.py
├── collector.py
└── env_loop.py
├── data
├── __init__.py
├── batch.py
├── batch_sampler.py
├── dataset.py
├── episode.py
├── segment.py
└── utils.py
├── envs
├── __init__.py
├── atari_preprocessing.py
├── env.py
└── world_model_env.py
├── game
├── __init__.py
├── dataset_env.py
├── game.py
└── play_env.py
├── main.py
├── models
├── __init__.py
├── actor_critic.py
├── blocks.py
├── diffusion
│ ├── __init__.py
│ ├── denoiser.py
│ ├── diffusion_sampler.py
│ └── inner_model.py
└── rew_end_model.py
├── play.py
├── player
├── __init__.py
├── action_processing.py
└── keymap.py
├── process_denoiser_files.py
├── process_upsampler_files.py
├── spawn.py
├── trainer.py
└── utils.py
/.gitignore:
--------------------------------------------------------------------------------
1 | *.pyc
2 | *.zip
3 | checkpoints
4 | **/__pycache__/
5 |
6 |
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | MIT License
2 |
3 | Copyright (c) 2024 Eloi Alonso
4 | Copyright (c) 2025 Enigma Labs AI
5 |
6 | Permission is hereby granted, free of charge, to any person obtaining a copy
7 | of this software and associated documentation files (the "Software"), to deal
8 | in the Software without restriction, including without limitation the rights
9 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 | copies of the Software, and to permit persons to whom the Software is
11 | furnished to do so, subject to the following conditions:
12 |
13 | The above copyright notice and this permission notice shall be included in all
14 | copies or substantial portions of the Software.
15 |
16 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 | SOFTWARE.
23 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # Multiverse: The First AI Multiplayer World Model
2 |
3 | 🌐 [Enigma-AI website](https://enigma-labs.io/) - 📚 [Technical Blog](https://enigma-labs.io/) - [🤗 Model on Huggingface](https://huggingface.co/Enigma-AI/multiverse) - [🤗 Datasets on Huggingface](https://huggingface.co/datasets/Enigma-AI/multiplayer-racing-low-res) - 𝕏 [Multiverse Tweet](https://x.com/j0nathanj/status/1920516649511244258)
4 |
5 |
6 |
Two human players driving cars in Multiverse
7 |
8 |
9 |