[DO NOT MERGE] Testing v2.6 + pr2201#2513

Draft

KshitijLakhani wants to merge 3 commits intoNVIDIA:release_v2.6from

KshitijLakhani:klakhani/maint/release_v2.6_pr2201

Commits on Dec 12, 2025

[JAX] Remove unnecessary SWA calculation in _segment_ids_pos_to_seqlens_offsets() (#2201)

KshitijLakhani
and
pre-commit-ci[bot]
committed

Commits on Dec 16, 2025

Get seqlens and offsets in O(N) space instead of O(N*N) space
KshitijLakhani
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed