Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch - View it on GitHub