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

[build]
build
twine

[docs]
jupyter-book
sphinx

[tests]
pytest
pytest-cov
