[JAX] Support Ring Attention (Context Parallelism) #6171

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
JAX Python	22s
JAX C++	22s
PaddlePaddle C++	23s
PaddlePaddle Python	1m 1s
PyTorch C++	25s
PyTorch Python	2m 17s
	4m 50s