accelerate
beartype
einops>=0.7.0
pytorch-custom-utils>=0.0.11
torch>=2.0
tqdm
x-transformers>=1.27.3
