Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
476 Open 15,009 Closed
476 Open 15,009 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] Pointy stax cla: yes
#430 by j-towns was closed Oct 31, 2019 Loading…
4 tasks
7 👍 5
Static kwargs for jit cla: yes
#3532 by shoyer was closed Apr 9, 2021 Loading…
5 tasks done
1
19 👍 5
GMRES cleanup/fixup cla: yes pull ready Ready for copybara import and testing
#5034 by shoyer was merged Dec 3, 2020 Loading…
@shoyer
26 👍 5
ahead-of-time compilation cla: yes
#6034 by froystig was closed Nov 29, 2021 • Draft
@froystig
4 👍 5
experiment: add jax.typing module and use it in jax.lax pull ready Ready for copybara import and testing
#12018 by jakevdp was closed Oct 14, 2022 • Draft
@jakevdp
15 👍 5
WIP: tree vectorizing transformation cla: yes pull ready Ready for copybara import and testing
#3263 by shoyer was closed Nov 10, 2021 Loading…
4 of 18 tasks
1
6 👍 4
fix custom_jvp/vjp closure issues, and nondiff_argnums too! cla: yes JEP JAX enhancement proposal pull ready Ready for copybara import and testing
#4008 by mattjj was closed Oct 21, 2020 Loading…
5 tasks done
7
9 👍 4
BFGS algorithm cla: yes
#3101 by Joshuaalbert was merged Jul 29, 2020 Loading…
151 👍 3
WIP: add experimental sparse support cla: yes
#4422 by jakevdp was closed Apr 16, 2021 • Draft
@jakevdp
2 👍 3
cache tracing of (sub)calls when forming a jaxpr
#9181 by mattjj was closed Jul 20, 2022 • Draft
1 task
1
@mattjj
2 👍 3
Change the default matmul precision in JAX to highest precision. cla: yes
#7859 by copybara-service bot was closed May 9, 2024 Loading…
@hawkinsp
8 👍 3
Add GRU & LSTM layer to stax cla: yes
#2298 by RobertTLange was closed Apr 12, 2021 Loading…
16 👍 2
remove safe_mul (undo #383, also cf. #1052) cla: yes useful read PR or issue that contains useful design discussion
#2447 by mattjj was merged Mar 18, 2020 Loading…
1
1 👍 2
omnistaging on by default cla: yes
#4038 by mattjj was merged Sep 15, 2020 Loading…
3
8 👍 2
Curry jit, grad, value_and_grad and vmap. cla: yes
#3736 by BuddenD was closed Jul 24, 2020 Loading…
3 👍 2
WIP: MLIR/TACO-like sparse representation cla: yes
#6555 by jakevdp was closed Mar 4, 2023 • Draft
@jakevdp
1 👍 2
jax.experimental.tree_math cla: yes pull ready Ready for copybara import and testing
#8504 by shoyer was closed Dec 29, 2021 Loading…
1
@mattjj
19 👍 2
Switch jaxlib to use nanobind instead of pybind11.
#16100 by copybara-service bot was merged Aug 24, 2023 Loading…
6 👍 2
improve (and shorten!) pmap errors about inconsistent axis sizes pull ready Ready for copybara import and testing
#12800 by mattjj was merged Oct 21, 2022 Loading…
@mattjj
1 👍 2
Add linear_sum_assignment
#16974 by odneill was closed Nov 6, 2023 Loading…
1
@jakevdp
13 👍 2
pandax: experimental JAX pandas wrapper
#12329 by jakevdp was closed Nov 3, 2023 • Draft
@jakevdp
1 👍 2
Add axis parameter to lax.scan cla: no
#4591 by alonfnt was closed Aug 13, 2022 Loading…
1
@mattjj
8 👍 2
Add SVD to jax/numpy/linalg, jax/scipy/linalg cla: yes
#197 by vishwakftw was merged Jan 8, 2019 Loading…
30 👍 1
Adagrad optimizer cla: yes
#710 by rohan-anil was merged May 16, 2019 Loading…
5 👍 1
add a lot of systematic vmap tests cla: yes
#870 by mattjj was merged Jul 6, 2019 Loading…
4 tasks done
1
👍 1
ProTip! Follow long discussions with comments:>50.