Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
25 Open 3,279 Closed
25 Open 3,279 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

Build on Windows cla: yes pull ready Ready for copybara import and testing
#4843 by cloudhan was merged Nov 19, 2020 Loading…
1
79 ❤️ 12
Add Github Actions supported wheel-builds for CPU-only jaxlib on Winx64 pull ready Ready for copybara import and testing
#15009 by mlxd was merged Jun 12, 2023 Loading…
10 ❤️ 6
Differentiable scan! cla: yes
#704 by mattjj was merged May 13, 2019 Loading…
5 ❤️ 3
try simplifying random.multivariate_normal api cla: yes
#1515 by mattjj was merged Oct 22, 2019 Loading…
1
12 ❤️ 2
Consistently return JAX arrays instead of Numpy-classic arrays from j… cla: yes
#1192 by hawkinsp was merged Aug 16, 2019 Loading…
1 ❤️ 2
An implementation of an experimental syntactic sugar for 'for' and while loops and conditionals. cla: yes enhancement New feature or request
#1706 by gnecula was merged Nov 18, 2019 Loading…
28 ❤️ 2
Add ability to specify individual test targets via a regex cla: yes
#3549 by jakevdp was merged Jun 29, 2020 Loading…
14 ❤️ 2
Add gradients to the scatter_max and scatter_min operations. cla: yes
#3111 by alexdavies was merged May 19, 2020 Loading…
9 ❤️ 2
Attach source info to Jaxpr equations. cla: yes
#3421 by hawkinsp was merged Jun 17, 2020 Loading…
20 ❤️ 2
Remove some non-inclusive language cla: yes
#3710 by jakevdp was merged Jul 10, 2020 Loading…
1 ❤️ 2
DeviceArray.__iter__ returns DeviceArrays, without host sync cla: yes pull ready Ready for copybara import and testing
#3821 by mattjj was closed Oct 16, 2021 Loading…
1
10 ❤️ 2
[host_callback] Added API to call host computations and return values to the accelerator computation cla: yes pull ready Ready for copybara import and testing
#5243 by gnecula was merged Jan 6, 2021 Loading…
30 ❤️ 2
improve an escaped tracer error cla: yes pull ready Ready for copybara import and testing
#4663 by mattjj was merged Oct 21, 2020 Loading…
2 ❤️ 2
specialize tree prefix error message for list/tuple pull ready Ready for copybara import and testing
#14093 by mattjj was merged Jan 20, 2023 Loading…
❤️ 2
add a jax.Array analog that can contain extended dtypes pull ready Ready for copybara import and testing
#20266 by mattjj was merged Apr 6, 2024 Loading…
12 ❤️ 2
prototype unfettered element types in jaxpr arrays pull ready Ready for copybara import and testing
#11768 by mattjj was merged Aug 6, 2022 Loading…
4 of 8 tasks
10 ❤️ 2
Start a new TPU interpret mode for Pallas pull ready Ready for copybara import and testing
#25097 by jburnim was merged Feb 6, 2025 Loading…
24 ❤️ 2
Form a tree of concatenations in jax.numpy.concatenate. cla: yes
#920 by hawkinsp was merged Jun 25, 2019 Loading…
1
2 ❤️ 1
parallelization work-in-progress: automatic data-parallel MLP cla: yes
#538 by froystig was merged Mar 29, 2019 Loading…
3 ❤️ 1
make namedtuples transparent (act as pytree nodes) cla: yes
#736 by mattjj was merged May 20, 2019 Loading…
1
1 ❤️ 1
Add documentation on asynchronous dispatch. cla: yes
#813 by hawkinsp was merged Jun 4, 2019 Loading…
1
1 ❤️ 1
Add an explicit DeviceArray.delete method. cla: yes
#792 by hawkinsp was merged May 30, 2019 Loading…
1
1 ❤️ 1
lax.custom_linear_solve primitive cla: yes
#1402 by shoyer was merged Oct 21, 2019 Loading…
21 ❤️ 1
add support for returning aux data from grad cla: yes
#484 by mattjj was merged Mar 8, 2019 Loading…
2
3 ❤️ 1
vmap improvements better_errors Improve the error reporting cla: yes documentation enhancement New feature or request
#1613 by mattjj was merged Oct 31, 2019 Loading…
3
28 ❤️ 1
ProTip! Adding no:label will show everything without a label.