Add tutorial Jupyter notebooks for world-model training and Atari benchmark

[ParamThakkar123]

Motivation

• Provide runnable, user-facing tutorial notebooks that demonstrate how to train each supported world-model family (Dreamer, PlaNet/RSSM, JEPA, IRIS, Genie, DIAMOND, DiT) on representative environments/datasets and how to run the Atari benchmark with trained checkpoints.
• Make the notebooks discoverable from the documentation site and ensure docs builds do not execute expensive training/download cells by default.

Description

• Added a new tutorials landing page at docs/source/tutorials/index.md and registered it in the main docs toctree (docs/source/index.rst).
• Added per-model tutorial notebooks under docs/source/tutorials/notebooks/: dreamer_dmc_walker.ipynb, planet_gym_cartpole.ipynb, jepa_cifar10.ipynb, iris_atari_pong.ipynb, genie_tinyworlds_sonic.ipynb, diamond_atari_breakout.ipynb, dit_cifar10.ipynb, and atari_benchmark.ipynb.
• Enabled notebook rendering in Sphinx by adding nbsphinx to docs/source/conf.py and set nbsphinx_execute = "never" so notebooks are rendered but not executed during doc builds.
• Kept long-running training/evaluation cells commented out in notebooks and used small smoke-run configs in examples so users can opt in to run them locally.

Testing

• Ran python -m py_compile docs/source/conf.py to validate the Sphinx config which succeeded.
• Verified each notebook is valid JSON and that all code cells parse with ast.parse (checked via a small script), which succeeded.
• Validated torchwm.create_config(...) overrides used by the notebooks (e.g. Dreamer and JEPA examples) to ensure the tutorial override fields are accepted, which succeeded.
• Checked availability of the optional nbsphinx package as part of verification and reported it missing in the current environment (import nbsphinx failed), which is expected for environments that do not install docs extras and does not block the change.

---

Codex Task

[github-actions]

PR Preview Action v1.8.1
🚀 View preview at https://ParamThakkar123.github.io/torchwm/pr-preview/pr-171/
Built to branch gh-pages at 2026-06-07 06:39 UTC. Preview will be ready when the GitHub Pages deployment is complete.