numpy
torch
Pillow
scikit-learn
scipy
scikit-image