numpy
scikit-learn>=1.0
mvlearn
scipy>=1.7
matplotlib
seaborn
pandas
tensorly
joblib

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

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

[probabilistic]
jax~=0.2.20
numpyro
arviz
