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

add custom_jvp / vjp, delete custom_transforms cla: yes JEP JAX enhancement proposal
#2026 by mattjj was merged Mar 22, 2020 Loading…
3 of 4 tasks
1
15 🎉 13
Differentiable scan! cla: yes
#704 by mattjj was merged May 13, 2019 Loading…
5 🎉 5
No more tuples cla: yes
#1224 by mattjj was merged Aug 21, 2019 Loading…
4 🎉 5
Add Generative Modelling by Estimating Gradients of Data Distribution… cla: yes
#1137 by dvmazur was merged Aug 8, 2019 Loading…
3 🎉 4
multibackend jit cla: yes
#1211 by levskaya was merged Aug 26, 2019 Loading…
11 🎉 4
Hoist loop-invariant residuals out of scan in partial eval cla: yes
#1664 by mattjj was merged Nov 11, 2019 Loading…
1
7 🎉 4
Lazy sublanguage cla: yes useful read PR or issue that contains useful design discussion
#1668 by mattjj was merged Jan 8, 2020 Loading…
6 of 11 tasks
2
63 🎉 4
Initial version of gmap cla: yes
#4006 by apaszke was merged Aug 17, 2020 Loading…
1 🎉 4
add internal vmappable interface (part 1) cla: yes pull ready Ready for copybara import and testing
#8451 by mattjj was merged Nov 4, 2021 Loading…
4 🎉 4
[shard-map] move shmap out of experimental, add docstring pull ready Ready for copybara import and testing
#20124 by mattjj was closed Apr 22, 2025 Loading…
7 🎉 4
transposed convolution implementation cla: yes
#592 by levskaya was merged Apr 10, 2019 Loading…
3 🎉 3
Add adaptive-step ODE solver and VJP calculation with JAX ops cla: yes
#1157 by cellweaver was merged Aug 13, 2019 Loading…
13 🎉 3
improve prng compile times with loop rolling cla: yes
#1175 by mattjj was merged Aug 13, 2019 Loading…
1
6 🎉 3
Allow shapecheck of PixelCNN++ cla: yes
#2017 by juliuskunze was merged Feb 14, 2020 Loading…
40 🎉 3
Implement shapecheck for more primitives cla: yes
#1990 by juliuskunze was merged Jan 16, 2020 Loading…
28 🎉 3
Add TPU Driver to jaxlib cla: yes
#1673 by henrytansetiawan was merged Nov 13, 2019 Loading…
4 🎉 3
Add a handwritten ThreeFry2x32 CUDA kernel. cla: yes
#1756 by hawkinsp was merged Nov 24, 2019 Loading…
1
1 🎉 3
Make optimization_barrier a public lax API. pull ready Ready for copybara import and testing
#23466 by hawkinsp was merged Sep 6, 2024 Loading…
7 🎉 3
Add Cholesky, QR, and Triangular solve implementations. cla: yes
#105 by hawkinsp was merged Dec 13, 2018 Loading…
7 🎉 2
add soft_pmap, plus very rough draft of parallelize cla: yes
#917 by mattjj was merged Jun 26, 2019 Loading…
2 🎉 2
add advanced indexing support to jax index ops cla: yes
#671 by mattjj was merged May 6, 2019 Loading…
1
12 🎉 2
Add Gumbel to jax.random, and test cla: yes
#560 by j-towns was merged Apr 2, 2019 Loading…
1 🎉 2
Use square(x) instead of pow(x, 2) in div JVP. cla: yes
#1388 by hawkinsp was merged Sep 23, 2019 Loading…
1
🎉 2
Don't AD through max-subtraction in softmax cla: yes
#2260 by jekbradbury was merged Jun 4, 2020 Loading…
7 🎉 2
Add batch_group_count to conv_general_dilated. cla: yes
#2635 by hawkinsp was merged Apr 9, 2020 Loading…
3 🎉 2
ProTip! no:milestone will show everything without a milestone.