einops
torch
numpy
rotary_embedding_torch
