Skip to content

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

[JAX] Support Ring Attention (Context Parallelism)

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

Annotations

1 warning

PyTorch Python

succeeded Nov 7, 2024 in 1m 57s