numpy
scipy
scikit-learn
pytest
matplotlib
pandas
seaborn
tensorly
joblib
mvlearn
tqdm
setuptools

[all]
torch>=1.9.0
torchvision
pytorch-lightning
jax
numpyro
arviz

[deep]
torch>=1.9.0
torchvision
pytorch-lightning

[probabilistic]
jax
numpyro
arviz
