README / README.md
ParamTh's picture
Shorten the organization card
7f3508d verified
|
Raw History Blame Contribute Delete
1.8 kB
metadata
title: README
emoji: 馃寑
colorFrom: pink
colorTo: purple
sdk: static
pinned: false

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
Dreamer IRIS PlaNet
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.