Home » Oreps » Jax » All Pull Requests » #40785
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

[Roll forward -- Take 2] Optimize large JAX Mesh construction #40785

Compare code on GitHub
Merged
copybara-service[bot] opened 3:16am on September 18, 2026 wanted to merge 1 commit into google/jax main from
test_983759493

Pull Request #40785 commits

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