numpy
torch
lightning
torchmetrics[audio]
