Simillusion

Open tools for building, training and sharing world models.

• 🧠 Simulusion: our PyTorch library. Dreamer, PlaNet, IRIS, DIAMOND, Genie and more behind one API. pip install simulusion
• 📦 Pretrained models: below, each trained on a single 4 GB laptop GPU
• 🚀 End-to-end world-model platform: coming soon

from huggingface_hub import snapshot_download
from simulusion.world_model import load_world_model

wm = load_world_model(snapshot_download("simillusion/iris-atari-boxing"), "iris")

🌐 Website · 💻 GitHub · 📖 Docs · 📦 Models

Tell us what would make world models useful in your work: open an issue.