release v2.8 rocm#428

Open

VeeraRajasekhar wants to merge 30 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

Commits on Feb 2, 2026

Fixed Jax build/test issues
VeeraRajasekhar
committed

Commits on Feb 6, 2026

[CI][Pytorch] Add softmax_type support to fused attention backend selection
VeeraRajasekhar
committed

Commits on Feb 12, 2026

Commits on Feb 17, 2026

Addressed reviews
VeeraRajasekhar
committed

Commits on Feb 19, 2026

Adressed reviews
VeeraRajasekhar
committed

Commits on Feb 23, 2026