assoc-scan>=0.0.4
axial_positional_embedding>=0.3.10
einops>=0.8.0
einx>=0.3.0
hyper-connections>=0.3.11
Ninja
rotary-embedding-torch
tensordict
torch>=2.0
tqdm
transformers>=4.30.0
x-transformers
