Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
277 Open 8,151 Closed
277 Open 8,151 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

Update remat config name in docs
#26929 opened Mar 5, 2025 by danijar Loading…
❤️ 1
Bump kiwisolver from 1.4.5 to 1.4.7 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24016 opened Sep 30, 2024 by dependabot bot Loading…
Bump importlib-resources from 6.4.0 to 6.4.5 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24018 opened Sep 30, 2024 by dependabot bot Loading…
Bump kiwisolver from 1.4.5 to 1.4.8 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28350 opened Apr 28, 2025 by dependabot bot Loading…
Bump importlib-resources from 6.4.0 to 6.5.2 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28352 opened Apr 28, 2025 by dependabot bot Loading…
Bump fsspec from 2024.5.0 to 2024.10.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24429 opened Oct 21, 2024 by dependabot bot Loading…
Bump hypothesis from 6.102.4 to 6.122.3 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25354 opened Dec 9, 2024 by dependabot bot Loading…
Bump fonttools from 4.51.0 to 4.55.3 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25512 opened Dec 16, 2024 by dependabot bot Loading…
Bump the pip group across 1 directory with 2 updates dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28835 opened May 19, 2025 by dependabot bot Loading…
1
Bump fsspec from 2024.5.0 to 2025.5.1 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#29011 opened May 26, 2025 by dependabot bot Loading…
Fix to make jax.scipy.special.sph_harm to be close to scipy.special.sph_harm for multiple theta or phi
#20772 opened Apr 16, 2024 by rajasekharporeddy Loading…
1
9
Reverts changelist 796974894
#31489 opened Sep 2, 2025 by copybara-service bot Loading…
[Mosaic:TPU] Fix bug in generalizes when dealing with replicated offsets and different tilings
#31255 opened Aug 22, 2025 by copybara-service bot Loading…
Update jax.tree module docs
#24925 opened Nov 15, 2024 by 8bitmp3 • Draft
Add example[newton_method]
#19160 opened Jan 2, 2024 by linjing-lab Loading…
4
[ROCm] [ROCm] BUILD file fixes and linter run
#25129 opened Nov 26, 2024 by Ruturaj4 Loading…
1
disable remat HLO pass by default
#24750 opened Nov 7, 2024 by keshavb96 Loading…
1
add options to compute interpolation using a dot product and to select search algorithm
#27701 opened Apr 3, 2025 by mattbahr Loading…
1
1
chore: format markdown table in README
#22986 opened Aug 10, 2024 by glitched-w0rld Loading…
Implicitly restarted arnoldi cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft
4
hoist constants to parameters in jit-compiled functions cla: yes
#3238 opened May 29, 2020 by froystig • Draft
[mutable-arrays] linearize the state
#26176 opened Jan 29, 2025 by mattjj Loading…
Add a linearization rule for remat
#28902 opened May 21, 2025 by dfm Loading…
2
Add a workflow to run benchmarks
#25386 opened Dec 10, 2024 by juliagmt-google Loading…
1
Add an example for executing serialized function in cpp
#24413 opened Oct 20, 2024 by fenghuizhang Loading…
8
ProTip! Updated in the last three days: updated:>2025-09-29.