Skip to content

/ jax

New pull request New
92 Open 2,186 Closed
92 Open 2,186 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Adding np.sort_complex within lax_numpy.py
cla: no
#3722 opened Jul 11, 2020 by Vaibhavs10
1
Change image resize implementation to use a matmul per dimension.
cla: yes
#3720 opened Jul 11, 2020 by hawkinsp
Import profiler in jax/__init__.py
cla: yes
#3719 opened Jul 11, 2020 by tomhennigan • Approved
[jax2tf] First draft of converting sort_p using TF2XLA
cla: yes
#3713 opened Jul 10, 2020 by SIben
13
allow only hashable values passed as static args
cla: yes
#3712 opened Jul 10, 2020 by mattjj
Make vjp cotangent functions pytree-like
cla: yes
#3705 opened Jul 9, 2020 by NeilGirdhar • Approved
10
Allow custom_jvp and custom_vjp to decorate methods
cla: yes
#3704 opened Jul 9, 2020 by NeilGirdhar
3
[jax2tf] Refactor tests to increase coverage
cla: yes
#3700 opened Jul 9, 2020 by gnecula
@SIben
stack traces modulo jax
cla: yes
#3674 opened Jul 7, 2020 by froystig
5
A compilable version of np.repeat.
cla: yes
#3670 opened Jul 6, 2020 by jg8610 • Changes requested
26
Compare treedefs by `num_leaves` not `traversal_` in `tree_transpose`.
cla: yes
#3659 opened Jul 4, 2020 by tomhennigan
@hawkinsp
6
Continuing the experiment with building jaxlib with GitHub actions
cla: yes
#3648 opened Jul 3, 2020 by gnecula
Adding support for array split
cla: yes
#3639 opened Jul 2, 2020 by AdrienCorenflos • Draft
3
improve error message for jnp.pad pad_width array
cla: yes
#3627 opened Jul 1, 2020 by mattjj • Draft
1
add lax.associative_scan to docs
cla: yes
#3583 opened Jun 27, 2020 by mattjj
5
Improve jnp.sum's error messages for tuple inputs
cla: yes
#3575 opened Jun 26, 2020 by IgorWilbert
@mattjj
8
Add QHM and QHAdam to `optimizers.py`, improve documentation.
cla: yes
#3570 opened Jun 26, 2020 by alexalemi
2
Refactor tests to define common functionality in test_util
cla: yes
#3561 opened Jun 25, 2020 by jakevdp
2
sketch out all-gather primitive
cla: yes
#3551 opened Jun 25, 2020 by mattjj • Draft
Github actions jaxlib build experiment.
cla: yes
#3547 opened Jun 24, 2020 by hawkinsp • Draft
1
Add physical optimization example
cla: no
#3544 opened Jun 24, 2020 by 0b01
2
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
21
ProTip! What’s not been updated in a month: updated:<2020-06-11.
You can’t perform that action at this time.