numpy
pandas
torch
torchmetrics
tqdm
nmslib
lightning<=3.6.1
typing_extensions