Skip to content

/ jax

New pull request New
81 Open 1,789 Closed
81 Open 1,789 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels.
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Reductions won't accept lists anymore
cla: yes
#3118 opened May 16, 2020 by joaogui1
1
Fix a corner case in `repeat`.
cla: yes
#3117 opened May 16, 2020 by midfield
3
Experimental thick-restart Lancozs method for sparse eigenvalue problems
cla: yes
#3114 opened May 15, 2020 by shoyer
Correct notebooks for neural networks in docs
cla: yes
#3113 opened May 15, 2020 by Baschdl
Add gradients to the scatter_max and scatter_min operations.
cla: yes
#3111 opened May 15, 2020 by alexdavies
3
Fix gradient for jax.scipy.ndimage.map_coordinates
cla: yes
#3110 opened May 15, 2020 by shoyer
1
sharded_jit scratch
cla: yes
#3107 opened May 15, 2020 by skye • Draft
BFGS algorithm
cla: yes
#3101 opened May 15, 2020 by Joshuaalbert
43
Add set_sharding method to be used within sharded_jit.
cla: yes
#3100 opened May 14, 2020 by skye
1
Add sharded_jit translation rule.
cla: yes
#3099 opened May 14, 2020 by skye
Add lax implementation of np.isin() and np.in1d()
cla: yes
#3087 opened May 13, 2020 by jakevdp
1
Implement np.histogram_bin_edges and np.histogram
cla: yes
#3081 opened May 13, 2020 by jakevdp • Approved
1
Use lax.scan inside jnp.polyval
cla: yes
#3076 opened May 13, 2020 by shoyer • Approved
1 8
add erf and erfc rules
cla: yes
#3051 opened May 12, 2020 by jacobjinkelly
Try to put all the requirements in a separate file
cla: yes
#3034 opened May 11, 2020 by gnecula
2
initial draft of hard coded docstrings
cla: yes
#3023 opened May 9, 2020 by joaogui1 • Draft
@gnecula
11
Move jax.numpy's implementation into a private jax._src directory.
cla: yes
#3015 opened May 8, 2020 by hawkinsp • Draft
2
Clean up jax.ops namespace.
cla: yes
#3011 opened May 8, 2020 by hawkinsp • Approved
Added argument check to all primitives.
cla: yes
#3005 opened May 8, 2020 by gnecula
WIP: better cond batching
cla: yes
#2961 opened May 5, 2020 by jekbradbury • Draft
Implemented numpy setxor1d
cla: yes
#2943 opened May 3, 2020 by joschkabraun • Changes requested
2
Numpy union1d
cla: yes
#2941 opened May 3, 2020 by joschkabraun
4
Add support for buffer donation in `jit` and `pmap`.
cla: yes
#2936 opened May 3, 2020 by tomhennigan
45
Implement mask for some primitives + jit.
cla: yes
#2922 opened May 1, 2020 by JuliusKunze
35
efficient nested pmap prototype
cla: yes
#2893 opened Apr 29, 2020 by skye • Draft
1
ProTip! Add no:assignee to see everything that’s not assigned.
You can’t perform that action at this time.