Skip to content

Pull requests: jax-ml/jax

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

Adding value_and_jac* functions cla: yes P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#762 opened May 23, 2019 by matt-graham Loading…
23 👍 4
[RFC] Add solver diagnostics for cg and bicgstab cla: yes
#7825 opened Sep 6, 2021 by PhilipVinc • Draft
3 👍 3
pallas: add argument zeroize_outputs to pallas_call()
#19399 opened Jan 17, 2024 by geggo Loading…
👍 1
Add caveat about cond and vmap to example
#27203 opened Mar 17, 2025 by alik-git Loading…
👍 1
Add generic type hints to parallel
#10147 opened Apr 5, 2022 by cgarciae Loading…
👍 1
Update remat config name in docs
#26929 opened Mar 5, 2025 by danijar Loading…
👍 1
Specify init step size for odeint cla: yes
#2604 opened Apr 4, 2020 by jacobjinkelly • Draft
6 👍 1
BDF solver for stiff ODEs cla: yes
#3781 opened Jul 16, 2020 by skrsna Loading…
1 of 4 tasks
9 👍 1
Use modern packaging standards for setuptools
#15280 opened Mar 29, 2023 by eli-schwartz Loading…
5 👍 1
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12 👍 1
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.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 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 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 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 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…
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 fonttools from 4.59.2 to 4.60.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#32012 opened Sep 22, 2025 by dependabot bot Loading…
Enhance TPU Matmul RHS transpose fusion to handle intervening slices.
#31625 opened Sep 8, 2025 by copybara-service bot Loading…
Canonicalize to vclamps (for vllm fp8 quant, maybe other uses)
#31452 opened Aug 31, 2025 by copybara-service bot Loading…
docs: Add note about computation perturbation in debug.print/breakpoint
#26682 opened Feb 22, 2025 by alhridoy Loading…
1
8
[pallas:mgpu] warp specialized pipeline with special delay_release.
#31483 opened Sep 2, 2025 by copybara-service bot Loading…
Clarify wait_recv semantics
#30000 opened Jul 7, 2025 by tengyifei Loading…
3
ProTip! Follow long discussions with comments:>50.