numpy
scipy
scikit-learn
scikit-prox
matplotlib
pandas
seaborn
tensorly
joblib
mvlearn
tqdm
torch
pytorch-lightning

[all]
torch
torchvision
pytorch-lightning
jax
numpyro
arviz

[deep]
torch
torchvision
pytorch-lightning

[probabilistic]
jax
numpyro
arviz
