torch
matplotlib
numpy
Pillow
