google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

[pallas] Preserve out_avals memory spaces in `_mpmd_map_discharge_rule` #40783

Open

copybara-service[bot] opened 7:18pm on September 17, 2026 wants to merge 66 Δ into google/jax main from

test_983558282

copybara-service[bot]

Diff Delta:

About 99 Diff Delta/hour

Classified as: General

[pallas] Preserve out_avals memory spaces in _mpmd_map_discharge_rule

3 total changed files

Loading changes...

You are now leaving the Pull Request Diff Zone

Thus concludes today’s episode of “Just One Small Change”

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