Skip to content

Pull requests: jax-ml/jax

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

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
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
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
Add sample_image.
#30872 opened Aug 9, 2025 by carlosgmartin Loading…
4
feat(gh-13291): Add Pareto distribution functions: cdf, logcdf, logsf, sf, ppf, logppf, and corresponding tests
#26556 opened Feb 15, 2025 by Qazalbash Loading…
Implementation of scipy.ndimage.gaussian_filter1d and gaussian_filter
#15359 opened Apr 2, 2023 by renecotyfanboy Loading…
1
normalize/rearrange (kw)args by signature in jit
#10724 opened May 16, 2022 by YouJiacheng Loading…
1
Fill in jax.distributed.initialize() arguments from environment variables
#12415 opened Sep 19, 2022 by crowsonkb Loading…
1
Fix zoom state for next iteration in line search
#11090 opened Jun 14, 2022 by david-anton Loading…
1
Add scan to checkify docs and fix rendering of ErrorCategories in docs.
#12079 opened Aug 24, 2022 by LenaMartens Loading…
1
4
Added working example of forward-forward algorithm
#15348 opened Mar 31, 2023 by sowmen Loading…
[WIP, Proposal] Improve argument annotations interface
#10746 opened May 17, 2022 by JeppeKlitgaard • Draft
3
Handle degenerate eigenvalues in JVP rule for eigh cla: yes
#1665 opened Nov 11, 2019 by shoyer • Draft
1
11
skip fori_loops with zero trip count cla: yes
#3295 opened Jun 2, 2020 by mattjj Loading…
1
4
Run scalar integer index normalization at trace time
#10683 opened May 12, 2022 by lgeiger Loading…
1
3
Adds kwargable decorator cla: yes
#3461 opened Jun 16, 2020 by BuddenD Loading…
4
avals with names cla: yes
#4397 opened Sep 24, 2020 by jekbradbury • Draft
1
change grad(jnp.abs)(0.+0.j) convention to be 1.+0j pull ready Ready for copybara import and testing
#10800 opened May 23, 2022 by mattjj Loading…
1
Create a temporary directory if unspecified by profiler.trace
#16633 opened Jul 5, 2023 by NeilGirdhar Loading…
HMM EM example cla: yes
#2405 opened Mar 12, 2020 by mattjj • Draft
Update conv_transpose docstring to inform users about swapped spatial axes in kernel as compared to Keras
#16238 opened Jun 2, 2023 by SriRangaTarun Loading…
1
Documentation fix: typo in jax.lax.custom_linear_solve doc
#10836 opened May 26, 2022 by dpsimpson Loading…
Add support for creating float0 arrays. cla: yes
#4437 opened Oct 2, 2020 by tomhennigan Loading…
1
6
Prefix PRNG Part 2: implement 1D prefix PRNG cla: yes
#2181 opened Feb 6, 2020 by j-towns Loading…
17
ProTip! Filter pull requests by the default branch with base:main.