torch>=2.4
torchvision
torchmetrics
lightning>=2.4
hydra-core
omegaconf
loguru
tabulate
pandas
rich
requests-cache
prettytable
submitit
timm
zstandard
typer
transformers
pillow>=11.3.0
wandb
hydra-submitit-launcher
matplotlib
datasets>=2.20.0
pyarrow>=20.0.0
minari[hdf5]>=0.5.3
scikit-learn>=1.7.0
richuru
opt-einsum
pylance
opencv-python-headless
kornia

[deepspeed]
deepspeed

[dev]
pytest
pytest-split
coverage
pytest-cov
codecov
pre-commit
ruff
sphinx
sphinx-gallery
sphinx-book-theme
sphinxcontrib-bibtex
matplotlib
memory_profiler
myst-parser

[jax]
jax
flax>=0.10.0
optax

[swanlab]
swanlab>=0.8.0

[trackio]
trackio
gradio_client
