torch
scipy
numpy
torchmetrics
