matplotlib
numpy
absl-py
torch
torchvision
pathlib