release v2.8 rocm#428

Draft

VeeraRajasekhar wants to merge 21 commits intorelease_v2.8_rocmfrom

veergopu/release_v2.8_rocm

Commits on Jan 19, 2026

Changed VERSION to 2.8.0

ptrendx
authored and
VeeraRajasekhar
committed
[JAX] Remove import jax.extend.ffi (#2193)

phu0ngng
authored and
VeeraRajasekhar
committed
[PyTorch] Add sink attention support from cuDNN (#2148)

authored and
VeeraRajasekhar
committed
[QA] Add pytest xml report for all tests in qa folder that use pytest (#2169)

shengfangd
authored and
VeeraRajasekhar
committed
[JAX] Local-Amax for Current-Scaling (#2183)

authored and
VeeraRajasekhar
committed
[JAX] Restore Shardy Rule with CompoundFactor (#2167)

phu0ngng
authored and
VeeraRajasekhar
committed
[JAX] Update JAX version requirement in pyproject.toml (#2197)

phu0ngng
authored and
VeeraRajasekhar
committed
[PyTorch] Unpin version of onnxscript and onnxruntime (#2202)

pggPL
authored and
VeeraRajasekhar
committed
[JAX] Fix XML filename in the L0_jax_uniitest (#2205)

phu0ngng
authored and
VeeraRajasekhar
committed
[JAX] CollectiveGemm (#2166)

phu0ngng
authored and
VeeraRajasekhar
committed
[JAX] Add xml export for `test_multiprocessing_encoder` and `test_cgemm` (#2210)

phu0ngng
authored and
VeeraRajasekhar
committed
[JAX] Address tolerance check for current scaling dact dbias (#2211)

jberchtold-nvidia
authored and
VeeraRajasekhar
committed

Commits on Jan 20, 2026

[Core][PyTorch] NVFP4 recipe (#2177)

authored and
VeeraRajasekhar
committed
Fix the segfault in the nvfp4 quantization (#2214)

authored and
VeeraRajasekhar
committed
[PyTorch] Add FP8 attention with current scaling (#2012)

authored and
VeeraRajasekhar
committed
[JAX] Load modules during initialize for Norm and Act primitives (#2219)

authored and
VeeraRajasekhar
committed
Fix the cuBLAS workspace alignment (#2223)

authored and
VeeraRajasekhar
committed
[PyTorch] Set usages for linear op quantizers before forward (#2222)

timmoon10
authored and
VeeraRajasekhar
committed

Commits on Jan 22, 2026

Resolved conflicts
VeeraRajasekhar
committed

Commits on Jan 23, 2026