Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
316 Open 11,566 Closed
316 Open 11,566 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 tagging tracer and collect/inject transforms cla: yes
#1660 by jekbradbury was closed Jan 29, 2022 Loading…
6 🚀 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
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
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
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
Add support for multihost pmaps. cla: yes
#1376 by skye was merged Sep 26, 2019 Loading…
9 🚀 3
prepare to switch to new remat pull ready Ready for copybara import and testing
#11743 by mattjj was merged Aug 4, 2022 Loading…
4 🚀 3
Add an undocumented method on jit() functions to clear the function c… pull ready Ready for copybara import and testing
#11064 by hawkinsp was merged Jun 14, 2022 Loading…
1
🚀 3
Stop using generators for linear_util transformations pull ready Ready for copybara import and testing
#24873 by dougalm was merged Nov 14, 2024 Loading…
1 🚀 3
Use log1p when computing log(1 + x) or log(1 - x) cla: yes
#1200 by majnemer was merged Aug 16, 2019 Loading…
1 🚀 2
Add support for interpolation options in quantile() and percentile() cla: yes
#2610 by jakevdp was merged Apr 7, 2020 Loading…
1
2 🚀 2
ProTip! no:milestone will show everything without a milestone.