Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
274 Open 8,175 Closed
274 Open 8,175 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

Add experimental rematerialization decorator cla: yes useful read PR or issue that contains useful design discussion
#1749 by mattjj was merged Nov 28, 2019 Loading…
1
6 🎉 11
omnistaging cla: yes
#3370 by mattjj was merged Jul 30, 2020 Loading…
6 🎉 10
Initial commit of jax2iree cla: yes
#7425 by stellaraccident was closed Nov 29, 2021 • Draft
31 🎉 10
Add tagging tracer and collect/inject transforms cla: yes
#1660 by jekbradbury was closed Jan 29, 2022 Loading…
6 🎉 9
Implement overrides of NumPy's public API on JAX arrays cla: yes
#611 by shoyer was closed Oct 24, 2019 Loading…
8 🎉 7
omnistaging on by default cla: yes
#4038 by mattjj was merged Sep 15, 2020 Loading…
3
8 🎉 7
Add example of an adaptive-step ODE solver and adjoint sensitivities cla: yes
#1068 by duvenaud was merged Jul 30, 2019 Loading…
6 🎉 5
make tracers tree-pretty-print their contents cla: yes
#2591 by mattjj was merged Apr 3, 2020 Loading…
1
🎉 5
Add MegaBlox grouped matrix multiplication kernels for TPU.
#20462 by copybara-service bot was merged Mar 29, 2024 Loading…
2 🎉 5
Implement batched cholesky decomposition using LAPACK/Cusolver cla: yes
#1956 by hawkinsp was merged Jan 7, 2020 Loading…
2
🎉 3
device_put_replicated cla: no
#1562 by mattjj was closed Jan 14, 2020 Loading…
13 🎉 2
Use 64-bit integers for indexing if any tensor dimension exceeds 2^31… cla: yes
#2182 by hawkinsp was merged Feb 7, 2020 Loading…
1
1 🎉 2
Update JAX to use XLA hyperbolic functions. cla: yes
#2415 by srvasude was merged Mar 19, 2020 Loading…
7 🎉 2
Tagging, take two cla: yes
#1803 by jekbradbury was closed May 20, 2020 Loading…
6 🎉 2
An implementation of id_print with CustomCall cla: yes
#2739 by gnecula was closed May 24, 2020 • Draft
8 🎉 2
Upstream propagate interpreter from oryx to jax.experimental cla: yes
#4088 by sharadmv was closed Apr 14, 2022 Loading…
🎉 2
[JAX] Use PocketFFT for FFTs on CPU instead of Eigen. cla: yes
#4697 by hawkinsp was closed Oct 23, 2020 Loading…
1 🎉 2
Enable sharded_jit_test and xmap SPMD tests for GPU. cla: yes
#5939 by copybara-service bot was closed Mar 24, 2021 Loading…
🎉 2
[MLIR] Don't name all top-level MHLO functions "main".
#8900 by copybara-service bot was closed Dec 10, 2021 Loading…
🎉 2
Documentation improvements. cla: yes
#1046 by hawkinsp was merged Jul 21, 2019 Loading…
🎉 1
added maml tutorial notebook cla: no
#447 by ericjang was closed Feb 25, 2019 Loading…
1 🎉 1
Add notebook demonstrating autobatched-log-probs vmap use case. cla: yes
#361 by matthewdhoffman was merged Feb 12, 2019 Loading…
4 🎉 1
Update XLA release to include fix for wrong output bug. cla: yes
#202 by hawkinsp was merged Jan 7, 2019 Loading…
1
🎉 1
Enable the remaining einsum tests. cla: yes
#345 by hawkinsp was merged Feb 8, 2019 Loading…
🎉 1
enable jit+pmap by merging pxla.py and xla.py cla: yes
#977 by mattjj was merged Jul 5, 2019 Loading…
1
1 🎉 1
ProTip! no:milestone will show everything without a milestone.