torch>=2.0.0
scikit-learn>=1.0.0
numpy>=1.21.0
matplotlib>=3.4.0
prettypyplot>=0.10.0
pyyaml>=6.0
click>=8.0.0
tqdm>=4.60.0

[docs]
mkdocs-material
mkdocstrings[python]
pymdown-extensions
