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

[ignite]
ignite

[record-keeper]
record_keeper>=0.9.31.dev6
