Top jax issues

Huge difference in speed between first and subsequent runs

question
gnool picture gnool·8 Oct 2020·4 Comments
👍1

Compile error on updating to JAX-0.2.0

documentation question
Joshuaalbert picture Joshuaalbert·25 Sep 2020·7 Comments

Broken docs check

bug build
JuliusKunze picture JuliusKunze·3 Sep 2020·4 Comments

`jax.scipy.linalg.expm` causes an infinite loop inside two nested `fori_loop`/`scan`s.

available bug
C-J-Cundy picture C-J-Cundy·27 Aug 2020·3 Comments

vmap unexpected behavior: NaNs in gradients

question
cagrikymk picture cagrikymk·13 Aug 2020·4 Comments

Fastest implementation of an elementwise gradient?

application question
tawe141 picture tawe141·11 Aug 2020·7 Comments

differentiating through the odeint when using complex numbers results in type errors

bug
AnnahDo picture AnnahDo·7 Aug 2020·7 Comments

Bad interaction between batch trace and custom jvp trace

bug
brianwa84 picture brianwa84·5 Aug 2020·7 Comments

jax.numpy.tile inconsistent with np.tile (when 0 in repeats)

bug
kratsg picture kratsg·31 Jul 2020·1 Comments
👍1

Allow mask of strided ops

enhancement question
JuliusKunze picture JuliusKunze·29 Jul 2020·5 Comments
 Previous issues
Next issues