Home » Oreps » Jax » All Pull Requests » #40783
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] Preserve out_avals memory spaces in `_mpmd_map_discharge_rule` #40783

Compare code on GitHub
Open
copybara-service[bot] opened 7:18pm on September 17, 2026 wants to merge 66 Ξ” into google/jax main from
test_983558282

Pull Request Overview

  • Opened on September 18, 2026
  • Status Open
  • Commit count 1 with first commit September 18, 2026

Total Delta

66 Total Diff Delta

Open Days

Open 0 weekdays

Test Delta

36 Diff Delta in Test Files
Breakdown by Phase

How long has this pull request spent in each phase of its lifecycle?

Data pending calculation for pull request

Author avatar

[pallas] Preserve out_avals memory spaces in `_mpmd_map_discharge_rule`

[pallas] Preserve out_avals memory spaces in _mpmd_map_discharge_rule

No comments have been left on this PR.

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