Top jax issues

jitted function is slow with JAX 0.1.42

bug performance
fehiepsi picture fehiepsi·23 Aug 2019·5 Comments

Add autodiff support for `reduce_window`

enhancement
brianwa84 picture brianwa84·20 Aug 2019·4 Comments

`is` keyword is not preserved under vmap.

documentation
sschoenholz picture sschoenholz·3 Aug 2019·3 Comments

multi-machine allreduce

enhancement
christopherhesse picture christopherhesse·10 Jul 2019·14 Comments
👍1

jitting grad gives wrong result in CPU

bug
fehiepsi picture fehiepsi·27 Jun 2019·4 Comments

jax.numpy.linalg ops missing GPU implementation

enhancement
romanngg picture romanngg·8 Jun 2019·7 Comments

Gradient of `np.exp` sometimes causes invalid values

bug
DylanMuir picture DylanMuir·27 May 2019·3 Comments

Cholesky behavior mismatch between jax and scipy

bug
lukemetz picture lukemetz·27 May 2019·4 Comments

Idiomatic Weight Sharing in Jax

c0nn3r picture c0nn3r·6 May 2019·2 Comments

jacfwd gives different results from jacrev

bug
proteneer picture proteneer·30 Apr 2019·5 Comments
👍1
 Previous issues
Next issues