[ ]
https://github.com/lucidrains/ring-attention-pytorch
[ ]
https://github.com/pytorch/pytorch/blob/0c8bb6f70c65b0a68fcb282cc1605c79ca5dabce/test/dynamo/test_triton_kernels.py#L628-L661