README.md

---
title: README
emoji: 🌀
colorFrom: pink
colorTo: purple
sdk: static
pinned: false
---

# Simillusion

**Open tools for building, training and sharing world models.**

- 🧠 **[Simulusion](https://github.com/ParamThakkar123/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

<table>
<tr>
<td width="33%"><a href="https://huggingface.co/simillusion/dreamer-dmc-cartpole-swingup"><img src="https://huggingface.co/simillusion/dreamer-dmc-cartpole-swingup/resolve/main/dreamer_dream.gif" alt="Dreamer" width="100%"></a></td>
<td width="33%"><a href="https://huggingface.co/simillusion/iris-atari-boxing"><img src="https://huggingface.co/simillusion/iris-atari-boxing/resolve/main/iris_dream.gif" alt="IRIS" width="100%"></a></td>
<td width="33%"><a href="https://huggingface.co/simillusion/planet-dmc-cartpole-swingup"><img src="https://huggingface.co/simillusion/planet-dmc-cartpole-swingup/resolve/main/planet_dream.gif" alt="PlaNet" width="100%"></a></td>
</tr>
</table>

```python
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](https://simulusion.vercel.app) · 💻 [GitHub](https://github.com/ParamThakkar123/simulusion) · 📖 [Docs](https://paramthakkar123.github.io/simulusion/) · 📦 [Models](https://huggingface.co/collections/simillusion/simulusion-demo-models-6ae0419992e0177bc1695362)

Tell us what would make world models useful in your work: [open an issue](https://github.com/ParamThakkar123/simulusion/issues).