numpy
scikit-learn
torch
torchvision
easy-module-attribute-getter
record-keeper
tensorboard
matplotlib
pretrainedmodels
pytorch-metric-learning
pandas
ax-platform
faiss-gpu
