absl-py
matplotlib
numpy
torch>=1.5.0
torchvision>=0.6.0
