Pull requests: jax-ml/jax
Pull requests list
Make xmapped functions with no axis resources callable more than once
cla: yes
pull ready
Ready for copybara import and testing
#5362
by apaszke
was merged Jan 15, 2021
Loading…
Replace scalar np.nonzero test with (1,) tests for NumPy 2.0
#20810
by apaszke
was closed Apr 22, 2024
Loading…
ProTip!
Updated in the last three days: updated:>2025-09-29.