Home » Oreps » Jax » All Pull Requests » #41854
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Pull Requests
Directory Browser
Velocity History
Releases

[Pallas:MGPU] Support transposing the swizzled dimension in `UnswizzleRef`. #41854

Compare code on GitHub
Closed
copybara-service[bot] opened 11:28am on October 21, 2026 wanted to merge 226 Δ into google/jax main from
test_1003935315

Pull Request #41854 commits

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy