deepspeed==0.8.1
faiss-gpu==1.7.2
numpy==1.24.2
ofa==0.1.0.post202203231606
optuna==3.1.0
Pillow==9.4.0
pytorch_lightning==1.9.0
scikit-image==0.20.0
torch==1.13.1
torchmetrics==0.11.1
torchvision==0.14.1
tqdm==4.64.1
ipykernel==6.21.1
matplotlib==3.6.3