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

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

[docs]
sphinx
sphinxawesome-theme
sphinx-gallery
sphinx-book-theme
matplotlib
memory_profiler
sphinx_design
myst-parser
