numpy
torch
openunmix
matplotlib
pillow