torch
numpy
matplotlib