librosa
rotary_embedding_torch
einops