Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
308 Open 11,519 Closed
308 Open 11,519 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
improve tree mismatch error in tree_map pull ready Ready for copybara import and testing
#23771 opened Sep 19, 2024 by mattjj Loading…
2 tasks
5 👍 2
pallas: add argument zeroize_outputs to pallas_call()
#19399 opened Jan 17, 2024 by geggo 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
Use modern packaging standards for setuptools
#15280 opened Mar 29, 2023 by eli-schwartz Loading…
5 👍 1
Add caveat about cond and vmap to example
#27203 opened Mar 17, 2025 by alik-git Loading…
👍 1
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12 👍 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
[TSAN CI] Try NO_CYTHON_COMPILE=true when installing Cython
#28967 opened May 23, 2025 by vfdev-5 • Draft
[CI] Introduce action lint workflow
#29355 opened Jun 10, 2025 by MichaelHudgins • Draft
[Mosaic:TPU] Better tpu.memref_reshape support for packed types by improving the reshaped layout inference
#31418 opened Aug 28, 2025 by copybara-service bot Loading…
Testing TPU CI (B)
#31137 opened Aug 18, 2025 by copybara-service bot Loading…
Update jax.grad docs
#24924 opened Nov 15, 2024 by 8bitmp3 • Draft
Uses include to reference xla packages in jaxlib xla BUILD
#28304 opened Apr 25, 2025 by yliu120 Loading…
Remove old ROCm build code kokoro:force-run pull ready Ready for copybara import and testing
#31381 opened Aug 27, 2025 by charleshofer Loading…
Test PR for runner
#30315 opened Jul 18, 2025 by quoctruong • Draft
[mutable-arrays] systematic scan ad test
#29834 opened Jun 28, 2025 by mattjj • Draft
Ignore more XLA flags for the JAX cache.
#22624 opened Jul 24, 2024 by nouiz Loading…
1
[DOC] Add documentation for jax_debug_infs configuration option to JAX debugging flags
#27012 opened Mar 8, 2025 by sumana-2705 Loading…
2
Update jax.numpy docs
#24920 opened Nov 15, 2024 by 8bitmp3 • Draft
print literal types in jaxprs pull ready Ready for copybara import and testing
#21415 opened May 24, 2024 by mattjj Loading…
try giving outside_call_p in/outfeed effects, tests fail
#19968 opened Feb 24, 2024 by mattjj • Draft
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.