Skip to content

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

[JAX] Support Ring Attention (Context Parallelism)

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

Annotations

1 warning

PyTorch C++

succeeded Nov 8, 2024 in 23s