Skip to content

/ jax

New pull request New
84 Open 1,878 Closed
84 Open 1,878 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
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
Added support for np.diagflat
cla: yes
#3259 opened May 30, 2020 by gsp-27
1
pmap(in_axes=None) of sharded_jit
cla: yes
#3257 opened May 29, 2020 by jekbradbury
Allow sharding infeed inside sharded_jit.
cla: yes
#3256 opened May 29, 2020 by skye • Approved
1
Improve dtype test coverage for random_test
cla: yes
#3254 opened May 29, 2020 by jakevdp
Add some type annotations to core and partial_eval.
cla: yes
#3251 opened May 29, 2020 by hawkinsp
Improve speed of tracing dynamic_update_slice
cla: yes
#3247 opened May 29, 2020 by j-towns
6
Skip known invars and outvars in JaxprTrace.process_call
cla: yes
#3242 opened May 29, 2020 by apaszke
Implementation of jax_to_tf.while
cla: yes
#3241 opened May 29, 2020 by gnecula
1
moved check_jaxpr code around to match eval_jaxpr
cla: yes
#3240 opened May 29, 2020 by mattjj
2
make pmap axis checking an exception, hoist
cla: yes
#3239 opened May 29, 2020 by mattjj • Approved
hoist constants to parameters in jit-compiled functions
cla: yes
#3238 opened May 29, 2020 by froystig • Draft
Don't keep primal arguments and results in the linearized jaxpr
cla: yes
#3233 opened May 28, 2020 by apaszke
Initial version of invertible AD implementation
cla: yes
#3232 opened May 28, 2020 by apaszke
Make ad_util.zero a class that carries avals (similar to UndefinedPrimal)
cla: yes
#3222 opened May 27, 2020 by apaszke • Approved
5
`odeint(f)` fails when `f` has a custom jvp
cla: yes
#3212 opened May 26, 2020 by jacobjinkelly
@mattjj
WIP: make standard abstract eval rules keep weak_type
cla: yes
#3175 opened May 21, 2020 by mattjj • Draft
1
unrolled and block-unrolled scan
cla: yes
#3159 opened May 20, 2020 by froystig
5
Enable XLA SPMD partitioning by default.
cla: yes
#3151 opened May 19, 2020 by skye • Draft
2
Support preconditions on scatter indices
cla: yes
#3147 opened May 19, 2020 by jekbradbury • Approved
4
Reductions won't accept lists anymore
cla: yes
#3118 opened May 16, 2020 by joaogui1 • Changes requested
9
Experimental thick-restart Lancozs method for sparse eigenvalue problems
cla: yes
#3114 opened May 15, 2020 by shoyer • Changes requested
@jakevdp
8
BFGS algorithm
cla: yes
#3101 opened May 15, 2020 by Joshuaalbert
77
Use lax.scan inside jnp.polyval
cla: yes
#3076 opened May 13, 2020 by shoyer • Approved
1 10
Try to put all the requirements in a separate file
cla: yes
#3034 opened May 11, 2020 by gnecula • Approved
@gnecula
6
initial draft of hard coded docstrings
cla: yes
#3023 opened May 9, 2020 by joaogui1 • Draft
@gnecula
12
ProTip! Type g i on any issue or pull request to go back to the issue listing page.
You can’t perform that action at this time.