scipy
matplotlib
pytorch
