torch>=1.13.0
torchvision>=0.14.0
torchaudio>=2.10.0
einops>=0.8.2
pyyaml>=6.0.3
tqdm>=4.67.1
opencv-python>=4.12.0.88
requests>=2.32.0
click>=8.0.0
gym>=0.26.2
gymnasium>=1.2.2
moviepy>=2.2.1
h5py>=3.11.0
plotly>=6.5.0
ale-py>=0.11.2
selenium>=4.41.0
scikit-learn>=1.8.0
umap-learn>=0.5.11
huggingface-hub>=1.14.0
playwright>=1.60.0
jax>=0.10.1

[all]
torchwm[brax,dev,docs,gym,ml,ml-agents,mujoco,robotics,viz]

[brax]
brax>=0.13.0

[dev]
pytest>=9.0.2
pytest-cov>=7.1.0
pre-commit>=4.5.0
mypy>=1.19.1

[docs]
sphinx<9.0.0,>=8.0.0
sphinx-autodoc-typehints<1.26.0
sphinx-copybutton>=0.5.2
sphinx-rtd-theme>=3.1.0
sphinxcontrib-bibtex>=2.6.5
sphinxcontrib-mermaid>=0.9.0
sphinx-thebe>=0.3.0
sphinxext-opengraph>=0.13.0
myst-parser<5.0.0,>=2.0.0
nbsphinx>=0.9.8

[gym]
huggingface-hub>=0.23.0
pygame>=2.6.1
autorom>=0.6.1

[ml]
tensorboard>=2.20.0
tensorboardx>=2.6.4
wandb>=0.16.0

[ml-agents]
mlagents-envs>=0.28.0

[mujoco]
mujoco>=3.2.0

[robotics]
gymnasium-robotics>=1.3.0

[viz]
fastapi[standard]>=0.116.0
uvicorn>=0.35.0
myst-parser<5.0.0,>=2.0.0
nbsphinx>=0.9.8
