scipy
numpy
torch
tensorboard
torchmetrics
piq
