einops>=0.3
rotary-embedding-torch
torch>=1.6
