Skip to content

Pull requests: jax-ml/jax

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

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 👀 1
Checkify: add debug mode with all checks+errors thrown automatically pull ready Ready for copybara import and testing
#12597 opened Sep 30, 2022 by LenaMartens Loading…
5 tasks
👀 1
WIP: better cond batching cla: yes
#2961 opened May 5, 2020 by jekbradbury • Draft
👀 1
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12 👀 1
Update jax.grad docs
#24924 opened Nov 15, 2024 by 8bitmp3 • Draft
Remove broken jax/jaxlib version test. pull ready Ready for copybara import and testing
#28515 opened May 5, 2025 by hawkinsp Loading…
Need to raise rtol/atol for Mosaic attn bfloat16
#28404 opened Apr 29, 2025 by hanzlfs Loading…
1
[mutable-arrays] systematic scan ad test
#29834 opened Jun 28, 2025 by mattjj • Draft
Make primary sidebar open by default and toggleable.
#28313 opened Apr 27, 2025 by zhangqiaorjc Loading…
1
jax profiler - trace events with absolute timestamps #26288
#26873 opened Mar 1, 2025 by Utkarsh-Aggarwal Loading…
1
[TSAN CI] Try NO_CYTHON_COMPILE=true when installing Cython
#28967 opened May 23, 2025 by vfdev-5 • Draft
Updated jax_debug_nans section in flags.md for clarity #17722
#27321 opened Mar 21, 2025 by Rohit-Singh-01 Loading…
1
[Mosaic] Add sc_subcore_parallel dimension semantic.
#31061 opened Aug 15, 2025 by copybara-service bot Loading…
Add Lineax link to jnp.linalg.solve
#17310 opened Aug 26, 2023 by patrick-kidger • Draft
feat(gh-13291): Add Pareto distribution functions: cdf, logcdf, logsf, sf, ppf, logppf, and corresponding tests
#26556 opened Feb 15, 2025 by Qazalbash Loading…
[Shmap/PartialAuto] Temporary solution for debug.print inside a partial-auto shard map.
#25705 opened Jan 1, 2025 by yliu120 Loading…
1
DOC: Update gpu_performance_tips.md
#23615 opened Sep 13, 2024 by Aiemu Loading…
Fix Dropout for mode = 'test' case in jax/example_libraries/stax.py file
#28086 opened Apr 17, 2025 by rajasekharporeddy Loading…
1
Add wheel build and py_import tests in JAX presubmit.
#31045 opened Aug 14, 2025 by copybara-service 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…
Measure the new presubmit.
#30865 opened Aug 9, 2025 by copybara-service bot Loading…
Add an example for executing serialized function in cpp
#24413 opened Oct 20, 2024 by fenghuizhang Loading…
8
[Mosaic:TPU][Relayouts] Fully replicated layouts are easy for masks, too
#31179 opened Aug 19, 2025 by copybara-service bot Loading…
Clarify the Applicability and Efficiency of Loss Function in autodiff_cookbook
#18677 opened Nov 27, 2023 by WenyanLiu Loading…
2
chore: format markdown table in README
#22986 opened Aug 10, 2024 by glitched-w0rld Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.