Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
553 Open 21,768 Closed
553 Open 21,768 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

Documentation fix: typo in jax.lax.custom_linear_solve doc
#10836 opened May 26, 2022 by dpsimpson Loading… updated May 26, 2022
snapshot the threefry PRNG implementation pull ready Ready for copybara import and testing
#11061 opened Jun 10, 2022 by froystig • Draft updated Jun 10, 2022
@froystig
Fix zoom state for next iteration in line search
#11090 opened Jun 14, 2022 by david-anton Loading… updated Jun 16, 2022
1
add more detail about aux_data in pytree docs cla: yes
#8739 opened Nov 30, 2021 by mattjj Loading… updated Jun 29, 2022
1
@mattjj
6
Maintain invariance to JIT and other transform when the function behavior depends on __class__ of inputs, e.g. functools.singledispatch
#11370 opened Jul 5, 2022 by YouJiacheng Loading… updated Jul 13, 2022
1
Added TPU precision idiosyncrasies to the Sharp Bits.
#11523 opened Jul 18, 2022 by charliexchen Loading… updated Jul 26, 2022
Typo in docs/debugging
#12144 opened Aug 29, 2022 by wonhyeongseo Loading… updated Aug 29, 2022
Add scan to checkify docs and fix rendering of ErrorCategories in docs.
#12079 opened Aug 24, 2022 by LenaMartens Loading… updated Sep 6, 2022
1
4
Scan as for
#12363 opened Sep 14, 2022 by sharadmv • Draft updated Sep 15, 2022
@sharadmv
Fill in jax.distributed.initialize() arguments from environment variables
#12415 opened Sep 19, 2022 by crowsonkb Loading… updated Sep 19, 2022
1
looking a djax-pjit stuff
#12439 opened Sep 21, 2022 by mattjj • Draft updated Sep 23, 2022
@mattjj
improve custom_jvp nondiff_argnums tracer error kokoro:force-run pull ready Ready for copybara import and testing
#12495 opened Sep 23, 2022 by mattjj Loading… updated Sep 23, 2022
Extend Common-Gotchas docs: speed of jnp.array
#11054 opened Jun 10, 2022 by onnoeberhard Loading… updated Oct 4, 2022
1
4
Add jet rules for scatter_mul and copy, and fix typos in scatter* docstrings.
#12483 opened Sep 23, 2022 by jacobjinkelly Loading… updated Oct 6, 2022
@mattjj
1
Upgrade the state primitive traceables to handle more forms of indexing
#12708 opened Oct 8, 2022 by sharadmv Loading… updated Oct 8, 2022
@sharadmv
1
Add PYTORCH_SAME padding to JAX. pull ready Ready for copybara import and testing
#12659 opened Oct 4, 2022 by RerRayne Loading… updated Oct 17, 2022
@apaszke
21
jnp.linalg.eigh: document symmetrize_input & error when using with UPLO pull ready Ready for copybara import and testing
#12696 opened Oct 7, 2022 by jakevdp • Draft updated Oct 18, 2022
1
@jakevdp
4
when i is int-like, make x.at[i].set(u) generate a DUS
#13002 opened Oct 27, 2022 by mattjj Loading… updated Oct 27, 2022
1 task
@mattjj
1
[RFC] Add solver diagnostics for cg and bicgstab cla: yes
#7825 opened Sep 6, 2021 by PhilipVinc • Draft updated Nov 3, 2022
3
Conditional scan JEP
#13184 opened Nov 10, 2022 by patrick-kidger Loading… updated Nov 11, 2022
@sharadmv
6
add scipy.special.factorial and scipy.special.comb pull ready Ready for copybara import and testing
#10948 opened Jun 2, 2022 by froystig • Draft updated Nov 15, 2022
@froystig
4
Allow passing empty path to compile cache
#13057 opened Nov 1, 2022 by lucasb-eyer Loading… updated Nov 24, 2022
@skye
1
fix broken notebook link pull ready Ready for copybara import and testing
#13607 opened Dec 11, 2022 by mattjj Loading… updated Dec 11, 2022
@mattjj
5
Implement fixed point routine jax.scipy.optimize.fixed_point
#12314 opened Sep 9, 2022 by JanLuca Loading… updated Dec 13, 2022
@froystig
4
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… updated Jan 11, 2023
5 tasks
ProTip! Updated in the last three days: updated:>2025-09-25.