Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
472 Open 15,120 Closed
472 Open 15,120 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

laplacian experiment pull ready Ready for copybara import and testing
#18902 opened Dec 10, 2023 by mattjj • Draft
@mattjj
23 ❤️ 1
Update remat config name in docs
#26929 opened Mar 5, 2025 by danijar Loading…
❤️ 1
feat(pallas): Optimize Pallas Attention + Benchmark
#17328 opened Aug 28, 2023 by jon-chuang Loading…
@sharadmv
17 ❤️ 1
Support mask(pmap) cla: yes
#4046 opened Aug 13, 2020 by j-towns • Draft
@mattjj
7 ❤️ 1
Adding the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18
Set default odeint tol appropriately for float32 cla: yes
#3444 opened Jun 15, 2020 by shoyer Loading…
1
2
Scan as for
#12363 opened Sep 14, 2022 by sharadmv • Draft
@sharadmv
Specify init step size for odeint cla: yes
#2604 opened Apr 4, 2020 by jacobjinkelly • Draft
6
feat(gh-13291): Add Pareto distribution functions: cdf, logcdf, logsf, sf, ppf, logppf, and corresponding tests
#26556 opened Feb 15, 2025 by Qazalbash Loading…
Add sample_image.
#30872 opened Aug 9, 2025 by carlosgmartin Loading…
4
Support building with Bzlmod
#32029 opened Sep 23, 2025 by meteorcloudy • Draft
2
Checkify: instrumentation context manager
#15389 opened Apr 4, 2023 by LenaMartens • Draft
11
Add regression test for hang when jit inputs are pruned.
#13703 opened Dec 17, 2022 by zhangqiaorjc Loading…
@zhangqiaorjc
HMM EM example cla: yes
#2405 opened Mar 12, 2020 by mattjj • Draft
add more options for jax_default_matmul_precision cla: yes
#6217 opened Mar 24, 2021 by mattjj Loading…
@mattjj
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12
custom transpose: nearly rewrite custom_vjp pull ready Ready for copybara import and testing
#10732 opened May 16, 2022 by froystig • Draft
@froystig
1
document class methods in autodiff autosummary pull ready Ready for copybara import and testing
#16639 opened Jul 6, 2023 by froystig Loading…
@froystig
1
wip exploring cache explanations pull ready Ready for copybara import and testing
#16338 opened Jun 9, 2023 by mattjj Loading…
@mattjj
Prefix PRNG Part 2: implement 1D prefix PRNG cla: yes
#2181 opened Feb 6, 2020 by j-towns Loading…
17
Add scan to checkify docs and fix rendering of ErrorCategories in docs.
#12079 opened Aug 24, 2022 by LenaMartens Loading…
1
4
Run scalar integer index normalization at trace time
#10683 opened May 12, 2022 by lgeiger Loading…
1
3
Support mask of broadcast, consts (arange, zeros, ...) cla: yes
#3472 opened Jun 17, 2020 by juliuskunze Loading…
Added doc: more cases in troubleshooting profiling
#16598 opened Jun 30, 2023 by OhadRubin Loading…
Better argnums/argnames inference
#10619 opened May 7, 2022 by JeppeKlitgaard • Draft
1 task
1
@hawkinsp
9
ProTip! Add no:assignee to see everything that’s not assigned.