accelerate>=0.25.0
beartype
classifier-free-guidance-pytorch>=0.4.2
einops>=0.7.0
ema-pytorch
gateloop-transformer>=0.1.5
pytorch-warmup
torch>=2.0
torch_geometric
torchtyping
tqdmvector-quantize-pytorch>=1.12.0
x-transformers>=1.26.0
