Closed cfoster0 closed 3 years ago
Should be a very easy addition. Existing JAX code for the 1D version of it is available here:
https://blog.eleuther.ai/rotary-embeddings/
Working on this in the rotary branch.
rotary
Should be a very easy addition. Existing JAX code for the 1D version of it is available here:
https://blog.eleuther.ai/rotary-embeddings/