Implement ring attention

[mawad-amd]

See algorithm and description here

https://arxiv.org/pdf/2310.01889