Pull requests: jax-ml/jax
Pull requests list
[mutable-arrays] fix indexing of effects in pjit typecheck
pull ready
Ready for copybara import and testing
#32434
by mattjj
was merged Oct 8, 2025
Loading…
Add
mosaic-gpu-plugin to the build script
#32433
by copybara-service
bot
was merged Oct 8, 2025
Loading…
out_sharding can be None for some primitives like rng_bit_generator_p.
#32427
by copybara-service
bot
was merged Oct 7, 2025
Loading…
Change python 3.13 to 3.14 in Bazel presubmit jobs.
#32421
by copybara-service
bot
was merged Oct 8, 2025
Loading…
Disable test that is breaking presubmits.
#32417
by copybara-service
bot
was merged Oct 7, 2025
Loading…
lax.acosh: express gradient using x^2-1 directly.
#32414
by copybara-service
bot
was merged Oct 7, 2025
Loading…
[pmap] Add note in
jax.pmap documentation pointing to migration docs.
#32413
by copybara-service
bot
was merged Oct 7, 2025
Loading…
Add an impl rule to empty2. Fixes https://github.com/jax-ml/jax/issues/32404
#32409
by copybara-service
bot
was closed Oct 7, 2025
Loading…
[jaxlib] Revived
PyTreeDef::FromNodeDataAndChildren
#32406
by copybara-service
bot
was merged Oct 7, 2025
Loading…
[pjrt] Do not ignore
shape layout in CommonPjRtClient::CreateViewOfDeviceBuffer
#32403
by copybara-service
bot
was closed Oct 7, 2025
Loading…
Add hermetic
tar and xz tools for extracting .tar and .tar.xz archives.
#32396
by copybara-service
bot
was merged Oct 8, 2025
Loading…
[TransferServer]: Allow resetting all outstanding buffer awaits and requests for buffers (sets
#32394
by copybara-service
bot
was closed Oct 7, 2025
Loading…
[pallas:mosaic_gpu] Allowed
scratch_shapes= to be a mapping in plgpu.kernel
#32388
by copybara-service
bot
was merged Oct 7, 2025
Loading…
Remove explicit opt-in lists for multi-TPU tests. Tag tests multiaccelerator instead.
#32384
by copybara-service
bot
was merged Oct 6, 2025
Loading…
Exclude tensorflow py_dep on free-threading and 3.14 builds
#32383
by copybara-service
bot
was merged Oct 7, 2025
Loading…
ProTip!
Filter pull requests by the default branch with base:main.