Top jax issues

Derivative of the determinant of a singular matrix is NaN

bug
shyoshyo picture shyoshyo·8 Mar 2020·26 Comments

Extremely slow runtime when using jacfwd()

performance question
JeffR1992 picture JeffR1992·24 Feb 2020·2 Comments
👍1

index-dependent scan function `lax.scani`

question
tachukao picture tachukao·24 Feb 2020·6 Comments

detach() or stop_gradient() equivalent?

question
anirbanl picture anirbanl·18 Jan 2020·2 Comments

isinstance(numpy.zeros(1), jax.numpy.ndarray) returns True

bug question
honnibal picture honnibal·16 Jan 2020·9 Comments

add maximally flatten option for tree_flatten

enhancement question
mgbukov picture mgbukov·30 Dec 2019·2 Comments

Add way to run jax program on CPU

question
jviereck picture jviereck·30 Oct 2019·6 Comments

How can I use jax to generate code for backward pass?

question
hfarazi picture hfarazi·24 Sep 2019·4 Comments
👍1

custom_transforms vjp rule clobbered under vmap

bug
mattjj picture mattjj·26 Aug 2019·2 Comments

jaxlib wheels aren't manylinux2010 compliant

bug
hzxie picture hzxie·13 Aug 2019·8 Comments
 Previous issues
Next issues