Skip to content

Pull requests: jax-ml/jax

New pull request New
724 Open 27,163 Closed
724 Open 27,163 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Be more defensive when converting 3.13-ft to 3.13t in build scripts.
#38458 opened Jun 13, 2026 by copybara-service Bot Loading…
@hawkinsp
Fix scipy.special.gammainc/gammaincc evaluation at a=inf.
#38457 opened Jun 13, 2026 by discobot Loading…
1
3
Fix spurious NaN in scipy.stats.chi2.logpdf at x=0 and x=inf when df=2.
#38456 opened Jun 13, 2026 by discobot Loading…
1
2
docs: clarify conv_transpose TensorFlow compatibility
#38455 opened Jun 13, 2026 by Mudassir55 Loading…
2
[remat3] start adding remat3-of-shmap kokoro:force-run pull ready Ready for copybara import and testing
#38453 opened Jun 13, 2026 by mattjj Collaborator • Draft
@mattjj
2
fix higher-order AD of custom_remat pull ready Ready for copybara import and testing
#38452 opened Jun 12, 2026 by mattjj Collaborator Loading…
@mattjj
Test nightly workflow
#38450 opened Jun 12, 2026 by psanal35 Contributor Loading…
1
Configure a portserver when running bazel tests.
#38446 opened Jun 12, 2026 by hawkinsp Collaborator Loading…
2
Add von Mises-Fisher 1D, 2D, and 3D samplers to random core
#38443 opened Jun 12, 2026 by egordon Loading…
5
[bug] fix jnp.isinf for non-standard float dtypes pull ready Ready for copybara import and testing
#38441 opened Jun 12, 2026 by jakevdp Collaborator Loading…
1
@jakevdp
2
Handle Gumbel ppf endpoint domains
#38436 opened Jun 12, 2026 by KirtiRamchandani Loading…
4
Handle Cauchy quantile domain boundaries
#38435 opened Jun 12, 2026 by KirtiRamchandani Loading…
3
Pallas emit_pipeline raises ValueError on empty grid
#38432 opened Jun 12, 2026 by vfdev-5 Member • Draft
3
Upgrade gRPC in XLA and TensorFlow to v1.81.0
#38431 opened Jun 12, 2026 by copybara-service Bot Loading…
fix(betaln): handle negative integer arguments via pole-cancellation closed form
#38428 opened Jun 12, 2026 by goingforstudying-ctrl Loading…
6
Allow remote references in _atomic_store_lowering_rule/_inline_mgpu_flat_transformed_args
#38427 opened Jun 12, 2026 by copybara-service Bot Loading…
Fix ragged_dot vmap batching rule for unbatched operands
#38424 opened Jun 12, 2026 by copybara-service Bot Loading…
take_current_trace should leave no trace kokoro:force-run pull ready Ready for copybara import and testing
#38415 opened Jun 11, 2026 by mattjj Collaborator Loading…
@mattjj
1
Add a jtu.is_libtpu_at_least version guard. kokoro:force-run pull ready Ready for copybara import and testing
#38407 opened Jun 11, 2026 by hawkinsp Collaborator Loading…
3
Make persistent compilation cache support host callbacks kokoro:force-run pull ready Ready for copybara import and testing
#38405 opened Jun 11, 2026 by skye-ant Loading…
@pschuh
3
Bump aws-actions/configure-aws-credentials from 6.1.0 to 6.2.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update Github_actions code
#38400 opened Jun 11, 2026 by dependabot Bot Loading…
Added debug arg to plgpu.kernel function
#38399 opened Jun 11, 2026 by vfdev-5 Member Loading…
3
Add a config flag to be able to blanket disabling of custom call safety checks in export.
#38393 opened Jun 11, 2026 by copybara-service Bot Loading…
Add bcomplex32 (complex<bfloat16>) ExtendedDType support
#38392 opened Jun 11, 2026 by DeanTMaxim Loading…
4 of 5 tasks
@jakevdp
6
Propagate NaNs in uniform stats functions
#38390 opened Jun 11, 2026 by KirtiRamchandani Loading…
1
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.