matplotlib
torch
Pillow
io