numpy
torch>=1.8.1
scipy
tqdm
PyYAML
click
matplotlib
ipywidgets
huggingface_hub
safetensors
tensorboard

[build]
build
twine

[docs]
jupyter-book
sphinx

[tests]
pytest
pytest-cov
