numpy
pandas
torch
torchmetrics
tqdm
faiss-cpu
lightning<=3.6.1
typing_extensions