implementing Rotary positional embeddings from scratch and trying to think through improvements/adaptation for other fields!