numpy
torch
torchvision
torchmetrics
pytorch-metric-learning>=1.0.0.dev5

[ignite]
pytorch-ignite

[lightning]
pytorch-lightning

[record-keeper]
record-keeper>=0.9.31.dev6
