accelerate==0.12.0
albumentations==1.2.0
docarray[common]>=0.13.25
jina-hubble-sdk>=0.15.2
mlflow==1.27.0
onnx==1.12.0
pydantic==1.9.1
pytorch-metric-learning==1.4.0
rich>=12.4.4
scipy==1.8.1
torchvision==0.13.0
torch==1.12.0
transformers==4.20.1
wandb==0.12.19
