numpy
scipy
numba
pot
torch
pytorch-lightning
