beartype
einops>=0.7.0
gateloop-transformer>=0.5.1
rotary-embedding-torch
torch>=2.1
