accelerate
beartype
einops>=0.6
ema-pytorch
fairseq
joblib
local-attention>=1.5.7
Mega-pytorch
scikit-learn
sentencepiece
torch>=1.12
torchaudio
transformers
tqdm
vector-quantize-pytorch>=0.10.15
