Skip to content

/ jax

New pull request New
83 Open 2,086 Closed
83 Open 2,086 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Instantiate a dropvar per outvar rather than using the same one
cla: yes
#3552 opened Jun 25, 2020 by sharadmv
sketch out all-gather primitive
cla: yes
#3551 opened Jun 25, 2020 by mattjj
Fix eigh JVP to ensure that both the primal and tangents of the eigen…
cla: yes
#3550 opened Jun 25, 2020 by hawkinsp • Approved
1 2
Add ability to specify individual test targets via a regex
cla: yes
#3549 opened Jun 24, 2020 by jakevdp
4
Github actions jaxlib build experiment.
cla: yes
#3547 opened Jun 24, 2020 by hawkinsp • Draft
Add physical optimization example
cla: no
#3544 opened Jun 24, 2020 by 0b01
2
Fix lazy broadcast issue
cla: yes
#3536 opened Jun 24, 2020 by j-towns • Approved
3
Static kwargs for jit
cla: yes
#3532 opened Jun 23, 2020 by shoyer 5 of 5
1
Add doubledouble exp() primitive
cla: yes
#3523 opened Jun 23, 2020 by jakevdp
@jakevdp
summary statistics of jaxpr equations
cla: yes
#3522 opened Jun 22, 2020 by froystig • Draft
Update JAX Quickstart with parallelization (`pmap`), other changes
cla: yes
#3520 opened Jun 22, 2020 by 8bitmp3
@skye
10
Add np.trim_zeros numpy function
cla: yes
#3499 opened Jun 21, 2020 by ayush-1506
2
[jax2tf] Add special case for translation of lax.gather to tf.gather.
cla: yes
#3486 opened Jun 18, 2020 by gnecula
Support mask of broadcast, consts (arange, zeros, ...)
cla: yes
#3472 opened Jun 17, 2020 by JuliusKunze
Adds kwargable decorator
cla: yes
#3461 opened Jun 16, 2020 by BuddenD
3
Set default odeint tol appropriately for float32
cla: yes
#3444 opened Jun 15, 2020 by shoyer
1 2
Fix init step sizes to account for the time span of integration
cla: yes
#3442 opened Jun 15, 2020 by samuela • Approved
@shoyer
1 11
link to nn libraries in readme
cla: no
#3422 opened Jun 12, 2020 by mattjj • Approved
@skye @mattjj @jekbradbury
3
Add jax.linear_transpose
cla: yes
#3398 opened Jun 10, 2020 by shoyer 0 of 2
2
Add variadic reduce (re-land #3342)
cla: yes
#3393 opened Jun 9, 2020 by jakevdp
@jakevdp
3
omnistaging
cla: yes
#3370 opened Jun 8, 2020 by mattjj • Draft
Add segment_max operator.
cla: yes
#3338 opened Jun 5, 2020 by alexdavies • Changes requested
7
Extract JAX primitive test harness from lax_test.py.
cla: yes
#3333 opened Jun 5, 2020 by gnecula
12
skip fori_loops with zero trip count
cla: yes
#3295 opened Jun 2, 2020 by mattjj
1 4
Tree vectorizing transformation
cla: yes
#3263 opened May 31, 2020 by shoyer 4 of 18
1 2
ProTip! Follow long discussions with comments:>50.
You can’t perform that action at this time.