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 Python

succeeded Nov 8, 2024 in 1m 54s