numpy
torch
backpack-for-pytorch
matplotlib
scipy