torch
einops
opt_einsum
rotary_embedding_torch
