rotary_embedding_torch
torch
jupyter
matplotlib