beartype
einops>=0.7.0
rotary-embedding-torch
torch>=1.6
