torch>=2.0
einops
rotary-embedding-torch
