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

[all]
protobuf==3.20.1
torch>=1.9.0
torchvision
pytorch-lightning
Pillow
jax~=0.2.20
numpyro
arviz

[deep]
protobuf==3.20.1
torch>=1.9.0
torchvision
pytorch-lightning
Pillow

[probabilistic]
jax~=0.2.20
numpyro
arviz
