Pull requests: jax-ml/jax
Pull requests list
Make convert_element_type_p a standard primitive and remove the usage of
utils.standard_abstract_eval.
#32348
opened Oct 4, 2025 by
copybara-service
bot
Loading…
[debug_info] More fixes in places where arg_names is incorrect
#31613
opened Sep 7, 2025 by
copybara-service
bot
Loading…
[debug] Extract reproducers from JAX errors.
pull ready
Ready for copybara import and testing
#31867
opened Sep 17, 2025 by
gnecula
Loading…
jnp.setdiff1d: handle non-1D inputs with assume_unique=True
kokoro:force-run
pull ready
Ready for copybara import and testing
#32340
opened Oct 3, 2025 by
jakevdp
Loading…
Use slice instead of iota for simple cases of vmap over Refs.
#32315
opened Oct 2, 2025 by
copybara-service
bot
Loading…
[jaxlib] Generate .pyi files for the native extensions via nanobind stubgen
#32287
opened Oct 2, 2025 by
copybara-service
bot
Loading…
Expose system memory buffer size for PM Sampling
#32265
opened Oct 1, 2025 by
copybara-service
bot
Loading…
[XLA:Python] Fix Python profiler hooks under Python free threading.
#32341
opened Oct 3, 2025 by
copybara-service
bot
Loading…
Support explicit integer padding in lax.conv_transpose
#32268
opened Oct 1, 2025 by
samanklesaria
Loading…
[Pallas] Add HiType support for pallas_call
#32320
opened Oct 3, 2025 by
copybara-service
bot
Loading…
Make jax-abstract-dunder-array a hard error.
#32318
opened Oct 2, 2025 by
copybara-service
bot
Loading…
ProTip!
Updated in the last three days: updated:>2025-10-01.