equinox
equinox copied to clipboard
RoPE Embeddings
Added RoPE embeddings from the RoFormer paper.
I need to add this to my transformer to perform some tests first before I can mark this as ready. Also if it's not needed/wanted, feel free to close this PR :)
If merged, closes #566