torch
torchvision
transformers
torchmetrics
lightning
hydra-core
loguru
tabulate
gymnasium[all]
minari
stable-pretraining
nevergrad
einops
datasets
pygame
pymunk
shapely

[dev]
pytest
coverage
pytest-cov
pytest-rich
codecov
pre-commit
ruff

[doc]
sphinx
sphinx-gallery
sphinx-book-theme
sphinxcontrib-bibtex
matplotlib
memory_profiler
furo
myst-parser
