numpy
scikit-learn
scipy>=1.5
matplotlib
seaborn
Pillow
pandas
tensorly
joblib

[deep]
torch>=1.7.0
torchvision
tensorboard

[probabilistic]
numpyro
arviz
