Skip to content

Pull requests: jax-ml/jax

New pull request New
752 Open 28,723 Closed
752 Open 28,723 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

[linalg] Fix eigh JVP division-by-zero on degenerate spectra and add precision tests (follow-up to #40149)
#40257 opened Aug 27, 2026 by SakshamKapoor2911 Loading…
[NVIDIA GPU] Add dynamic root to jax pbroadcast
#40253 opened Aug 27, 2026 by Tixxx Contributor Loading…
Propagate Mosaic GPU kernel launch errors to XLA instead of calling abort().
#40252 opened Aug 27, 2026 by copybara-service Bot Loading…
Make image resizing a primitive
#40251 opened Aug 27, 2026 by ahmedtaha100 Contributor Loading…
1
@hawkinsp
[mgpu] Add collective_axes parameter to try_cluster_cancel.
#40250 opened Aug 27, 2026 by copybara-service Bot Loading…
@chr1sj0nes
Pass closed-over constants as arguments to functions with inline=JAX_LATE.
#40249 opened Aug 27, 2026 by copybara-service Bot Loading…
@hawkinsp
[Mosaic GPU] add support to GPU interpret mode for more complicated transforms and oob filling
#40245 opened Aug 26, 2026 by copybara-service Bot Loading…
[Mosaic GPU] Improve error handling in GPU Interpret Mode
#40243 opened Aug 26, 2026 by copybara-service Bot Loading…
Strip memory spaces and synchronize aliased carry in Pallas HLO interpreter.
#40241 opened Aug 26, 2026 by copybara-service Bot Loading…
@cgarciae
Explicitly use LLVM19 in pytest 3.15 job.
#40240 opened Aug 26, 2026 by copybara-service Bot Loading…
@belitskiy
Refactor xla::ifrt::ArraySpec from a struct to a class
#40239 opened Aug 26, 2026 by copybara-service Bot Loading…
numpy: return empty results for intersect1d and setxor1d at size 0
#40231 opened Aug 26, 2026 by VaggelisGian Loading…
@jakevdp
4
numpy: return empty result from setdiff1d when size is zero
#40230 opened Aug 26, 2026 by VaggelisGian Loading…
@jakevdp
3
Match numpy output shapes in jnp.unique for empty arrays
#40229 opened Aug 26, 2026 by VaggelisGian Loading…
@jakevdp
2
scipy.optimize: fix signed bracket width in line search zoom
#40228 opened Aug 26, 2026 by VaggelisGian Loading…
scipy.optimize: report ftol termination in L-BFGS as converged
#40227 opened Aug 26, 2026 by VaggelisGian Loading…
Allow rank 3 cp.async
#40222 opened Aug 26, 2026 by samanklesaria Contributor Loading…
1
Manual Localization through Locality Domains
#40221 opened Aug 25, 2026 by keshavb96 Contributor • Draft
Fix Fusible.jvp to instantiate symbolic zero tangents.
#40218 opened Aug 25, 2026 by copybara-service Bot Loading…
Avoid redundant device_put for callback arguments already on CPU.
#40210 opened Aug 25, 2026 by copybara-service Bot Loading…
@jburnim
Bump rules_ml_toolchain to the latest version
#40205 opened Aug 25, 2026 by copybara-service Bot Loading…
@yuriivcs
1
Integrate StableHLO at openxla/stablehlo@ea56251e
#40204 opened Aug 25, 2026 by copybara-service Bot Loading…
Internal change
#40202 opened Aug 25, 2026 by copybara-service Bot Loading…
@rdyro
Integrate Triton up to a77e7c793abc0d0c923a9afb275058e2fe57a198
#40199 opened Aug 25, 2026 by copybara-service Bot Loading…
Align cdiv behavior for negative numbers between Python and JAX paths.
#40198 opened Aug 25, 2026 by copybara-service Bot Loading…
ProTip! What’s not been updated in a month: updated:<2026-07-27.