astropy
pytorch-ignite
toml
tomlkit
torch
torchvision
tensorboardX
tensorboard
GPUtil
schwimmbad
chromadb
holoviews
bokeh
jupyter_bokeh
datashader
pandas
numpy
scipy
cython
mlflow
pynvml
umap-learn
pooch

[dev]
asv==0.6.4
jupyter
matplotlib
mypy
pre-commit
pytest
pytest-cov
ruff

[examples]
