[typing] Avoid non-contiguous overloads in splash_attention_kernel.py #41856

Merged
stroxler opened 11:38am on October 21, 2026 wanted to merge 1 commit into google/jax main from
fix-noncontiguous-overload

Pull Request #41856 commits