Skip to content

Pull requests: jax-ml/jax

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

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
Checkify: instrumentation context manager
#15389 opened Apr 4, 2023 by LenaMartens • Draft
11
Added doc: more cases in troubleshooting profiling
#16598 opened Jun 30, 2023 by OhadRubin Loading…
Expose a way to check for a name primitive in JAX checkpoint API
#16578 opened Jun 28, 2023 by inoryy Loading…
Use modern packaging standards for setuptools
#15280 opened Mar 29, 2023 by eli-schwartz Loading…
5
optimize scan with constant-across-time outputs cla: yes
#1447 opened Oct 5, 2019 by jekbradbury • Draft
1
prefer operator.index over int
#10395 opened Apr 21, 2022 by YouJiacheng • Draft
13
ProTip! Mix and match filters to narrow down what you’re looking for.