v0.4.1

What's Changed

• Add --data_file option to bypass HuggingFace download in TinyWorlds by @ParamThakkar123 in #142
• Fixed Genie training and inference related bugs by @ParamThakkar123 in #143
• feat/cli by @ParamThakkar123 in #144
• Simplify docs navbar and fix Mermaid diagrams by @ParamThakkar123 in #149
• Add benchmark command to torchwm CLI by @ParamThakkar123 in #148
• docs: add CLI documentation by @ParamThakkar123 in #145
• Fix docs rendering (MathJax & Mermaid), add World Models study guide, and improve API reference by @ParamThakkar123 in #146
• Add vectorized env and RL harness unit tests; fix SimWorker seed handling by @ParamThakkar123 in #150
• Clean up TorchWM CLI entrypoint and make tools exports lazy by @ParamThakkar123 in #154
• Docs: switch to Sphinx-managed MathJax, add copy-to-clipboard, and expand environment docs by @ParamThakkar123 in #155
• Add Brax environment backend (BraxImageEnv) with docs, config, and tests by @ParamThakkar123 in #156
• Add native MuJoCo image environment support, docs, and extras by @ParamThakkar123 in #151
• Add optional windowed-attention pooling for latent action tokens by @ParamThakkar123 in #158
• Add Gymnasium Robotics backend support and automatic MuJoCo v2/v3 fallback by @ParamThakkar123 in #157
• Consolidate benchmarks into world_models/benchmarks and update docs by @ParamThakkar123 in #161
• ci: publish releases to PyPI and sync pyproject version by @ParamThakkar123 in #164
• Add CLI docs to navigation and relocate Brax backend into environments section by @ParamThakkar123 in #165
• Refactor public API for easier TorchWM usage by @ParamThakkar123 in #159

Full Changelog: v0.4.0...v0.4.1