Skip to content

Pull requests: jax-ml/jax

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

omnistaging cla: yes
#3370 by mattjj was merged Jul 30, 2020 Loading…
6 🚀 14
add L-BFGS optimizer cla: yes pull ready Ready for copybara import and testing
#6053 by Jakob-Unfried was merged Jun 2, 2021 Loading…
3 of 4 tasks
@shoyer
41 🚀 12
Add a GPU implementation of lax.linalg.eig.
#24663 by copybara-service bot was merged Nov 18, 2024 Loading…
@dfm
2 🚀 8
Add tagging tracer and collect/inject transforms cla: yes
#1660 by jekbradbury was closed Jan 29, 2022 Loading…
6 🚀 6
Switch JAX to use OpenXLA repository. pull ready Ready for copybara import and testing
#14475 by hawkinsp was merged Mar 13, 2023 Loading…
@hawkinsp
12 🚀 6
add jax_default_matmul_precision flag and context manager cla: yes pull ready Ready for copybara import and testing
#6143 by mattjj was merged Mar 24, 2021 Loading…
2 tasks done
1
13 🚀 5
AWN-enabled reduction over named axes in reverse-mode AD cla: yes
#6950 by copybara-service bot was merged Jul 8, 2021 Loading…
@jekbradbury
11 🚀 5
Add initial implementation of a run_state primitive pull ready Ready for copybara import and testing
#15149 by sharadmv was merged Apr 4, 2023 Loading…
17 🚀 5
Build on Windows cla: yes pull ready Ready for copybara import and testing
#4843 by cloudhan was merged Nov 19, 2020 Loading…
1
79 🚀 5
Add MegaBlox grouped matrix multiplication kernels for TPU.
#20462 by copybara-service bot was merged Mar 29, 2024 Loading…
2 🚀 5
Differentiable scan! cla: yes
#704 by mattjj was merged May 13, 2019 Loading…
5 🚀 4
Added associative_scan. cla: yes
#2170 by dpiponi was merged May 21, 2020 Loading…
@mattjj
8 🚀 4
Faster gradient rules for {numpy,scipy}.linalg.solve cla: yes
#2220 by shoyer was merged Feb 19, 2020 Loading…
1
2 🚀 4
Attach source info to Jaxpr equations. cla: yes
#3421 by hawkinsp was merged Jun 17, 2020 Loading…
20 🚀 4
Grad of int cla: yes pull ready Ready for copybara import and testing
#4039 by LenaMartens was merged Oct 1, 2020 Loading…
8 tasks done
49 🚀 4
omnistaging on by default cla: yes
#4038 by mattjj was merged Sep 15, 2020 Loading…
3
8 🚀 4
single-operand cond cla: yes
#2993 by froystig was merged May 15, 2020 Loading…
13 🚀 4
Complex SVD JVP rule cla: yes pull ready Ready for copybara import and testing
#5225 by frederikwilde was merged Jan 13, 2021 Loading…
@jakevdp
24 🚀 4
support implicit broadcasting in transpose rules cla: yes pull ready Ready for copybara import and testing
#6470 by mattjj was merged Apr 16, 2021 Loading…
1 task done
6
13 🚀 4
WIP: named call context manager cla: yes pull ready Ready for copybara import and testing
#8290 by sharadmv was closed Feb 22, 2022 • Draft
12 of 13 tasks
22 🚀 4
initial spmd groundwork cla: yes
#272 by mattjj was merged Jan 28, 2019 Loading…
1 🚀 3
Add a more direct implementation of device_put. cla: yes
#960 by hawkinsp was merged Jul 1, 2019 Loading…
1
🚀 3
multibackend jit cla: yes
#1211 by levskaya was merged Aug 26, 2019 Loading…
11 🚀 3
Batching rule for custom_linear_solve cla: yes
#2099 by danieldjohnson was merged Jan 30, 2020 Loading…
8 🚀 3
Add TPU Driver to jaxlib cla: yes
#1673 by henrytansetiawan was merged Nov 13, 2019 Loading…
4 🚀 3
ProTip! Add no:assignee to see everything that’s not assigned.