Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
74 Open 6,768 Closed
74 Open 6,768 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

Postrelease 0.7.2 kokoro:force-run pull ready Ready for copybara import and testing
#31844 by kanglant was merged Sep 16, 2025 Loading… updated Oct 2, 2025
1
jnp.cov: align behavior with NumPy 2.2 change kokoro:force-run pull ready Ready for copybara import and testing
#32308 by jakevdp was merged Oct 2, 2025 Loading… updated Oct 2, 2025
@jakevdp
[mutable arrays] remove internal aliases kokoro:force-run pull ready Ready for copybara import and testing
#32142 by jakevdp was merged Oct 2, 2025 Loading… updated Oct 2, 2025
@jakevdp
1
jnp.cov: return NaN for empty inputs kokoro:force-run pull ready Ready for copybara import and testing
#32305 by jakevdp was merged Oct 2, 2025 Loading… updated Oct 2, 2025
1
@jakevdp
fix(doc): no autoselect of attention implementation kokoro:force-run pull ready Ready for copybara import and testing
#32199 by emergenz was merged Oct 2, 2025 Loading… updated Oct 2, 2025
@emilyfertig
5
In jax.scipy.linalg.block_diag, form a single wider concatenation rat… kokoro:force-run pull ready Ready for copybara import and testing
#32302 by hawkinsp was merged Oct 2, 2025 Loading… updated Oct 2, 2025
1
cache convert_const_himutables kokoro:force-run pull ready Ready for copybara import and testing
#32228 by cgarciae was merged Oct 1, 2025 Loading… updated Oct 1, 2025
@mattjj
1
Add Sici Function kokoro:force-run pull ready Ready for copybara import and testing
#32052 by michaeleliot was merged Oct 1, 2025 Loading… updated Oct 1, 2025
1
@jakevdp
68
[doc] fix formatting in jax.export API docs documentation kokoro:force-run pull ready Ready for copybara import and testing
#32220 by jakevdp was merged Oct 1, 2025 Loading… updated Oct 1, 2025
@jakevdp
1
Warn on import for jax.lib.xla_bridge, xla_client, xla_extensions kokoro:force-run pull ready Ready for copybara import and testing
#32193 by jakevdp was merged Oct 1, 2025 Loading… updated Oct 1, 2025
@jakevdp
1
[Pallas] Improve error msg for out-of-bounds stores of blocks of outputs. kokoro:force-run pull ready Ready for copybara import and testing
#31825 by jburnim was merged Sep 30, 2025 Loading… updated Sep 30, 2025
3
Expose jax.export at top-level for consistency with other submodules kokoro:force-run pull ready Ready for copybara import and testing
#32160 by athulya-anil was merged Sep 30, 2025 Loading… updated Sep 30, 2025
1
@jakevdp
7
Bump actions/cache from 4.2.4 to 4.3.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update Github_actions code kokoro:force-run pull ready Ready for copybara import and testing
#32184 by dependabot bot was merged Sep 30, 2025 Loading… updated Sep 30, 2025
@jakevdp
[mutable arrays] deprecate old public aliases for new_ref and Ref. kokoro:force-run pull ready Ready for copybara import and testing
#32107 by jakevdp was merged Sep 26, 2025 Loading… updated Sep 26, 2025
@jakevdp
1
Share binary size threshold kokoro:force-run pull ready Ready for copybara import and testing
#20040 by keshavb96 was closed Sep 26, 2025 Loading… updated Sep 26, 2025
@hawkinsp
13
Reduce block sizes for GPUs with less shared memory CI Optional GPU Presubmit Label to flag PR to run additional GPU testing not in standard presubmits kokoro:force-run pull ready Ready for copybara import and testing
#32072 by olupton was merged Sep 26, 2025 Loading… updated Sep 26, 2025
Support different dbias shape in JAX CuDNN SDPA API kokoro:force-run pull ready Ready for copybara import and testing
#32018 by Cjkkkk was merged Sep 26, 2025 Loading… updated Sep 26, 2025
1
@mattjj
3
[docs] New tutorial on Tracing kokoro:force-run pull ready Ready for copybara import and testing
#30703 by melissawm was merged Sep 25, 2025 Loading… updated Sep 25, 2025
@emilyfertig
17
added solve_sylvester and accompanying tests kokoro:force-run pull ready Ready for copybara import and testing
#28810 by Dekermanjian was merged Sep 25, 2025 Loading… updated Sep 25, 2025
@jakevdp
145
Fix oob index errors for empty indices. kokoro:force-run pull ready Ready for copybara import and testing
#32108 by jakevdp was merged Sep 25, 2025 Loading… updated Sep 25, 2025
@jakevdp
3
[hijax] vmap is a primitive, behind a flag kokoro:force-run pull ready Ready for copybara import and testing
#32085 by mattjj was merged Sep 25, 2025 Loading… updated Sep 25, 2025
@mattjj
Reorganize the tutorials, take 2 documentation kokoro:force-run pull ready Ready for copybara import and testing
#31269 by melissawm was merged Aug 28, 2025 Loading… updated Sep 24, 2025
20
Simplify tree_reduce signature using Unspecified bare class. kokoro:force-run pull ready Ready for copybara import and testing
#30133 by carlosgmartin was merged Sep 24, 2025 Loading… updated Sep 24, 2025
@jakevdp
4
Update documentation to reflect experimental support for WSL2 on ROCm kokoro:force-run pull ready Ready for copybara import and testing
#32076 by JehandadKhan was merged Sep 24, 2025 Loading… updated Sep 24, 2025
1
Implement custom derivatives for hyp2f1 kokoro:force-run pull ready Ready for copybara import and testing
#31994 by mattbahr was merged Sep 23, 2025 Loading… updated Sep 23, 2025
1
@jakevdp
7
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.