Top jax issues

Tips for debugging NaNs in gradient?

bug
martiningram picture martiningram·4 Mar 2019·21 Comments
👍3

Custom GPU ops

proteneer picture proteneer·17 Apr 2019·22 Comments

Jax scans are slower than expected

performance
dionhaefner picture dionhaefner·23 Mar 2020·20 Comments

how to detect if GPU is being used? (feature request)

enhancement
murphyk picture murphyk·3 Jul 2019·3 Comments
👍1

namedtuple support in arguments to transformed functions

enhancement
jonasrauber picture jonasrauber·25 Feb 2019·6 Comments

Limit number of threads when running on CPU

documentation
dionhaefner picture dionhaefner·21 Oct 2019·22 Comments

lax.dynamic_slice inside jit

documentation question
shoyer picture shoyer·9 Jul 2019·7 Comments

inside pmap, grad(lambda x: psum(loss(x))) inconsistent with jvp(lambda x: psum(loss(x))

bug
sharadmv picture sharadmv·6 Aug 2020·13 Comments

Print return of jit method in class

question useful_read
rdaems picture rdaems·25 Oct 2019·6 Comments
👍1

Differentiable matrix-free linear algebra, optimization and equation solving

shoyer picture shoyer·20 Oct 2019·35 Comments
🎉6 🚀4
 Previous issues
Next issues