-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add nd_loop and Enable block_n tiling for all_gather_lhs_matmul
kokoro:force-run
pull ready
Ready for copybara import and testing
#29822
opened Jun 27, 2025 by
hanzlfs
Loading…
Retry connection errors to combat flakiness in google metadata server during tpu initialization (reopens #24432)
#26919
opened Mar 4, 2025 by
dlwh
Loading…
Update JAX README - add
shard_map information (pmap deprecation)
#17210
opened Aug 21, 2023 by
8bitmp3
Loading…
ProTip!
Updated in the last three days: updated:>2025-09-19.