Top jax issues

compiling time grows non-linear w.r.t. the number of random.split ops

bug performance
fehiepsi picture fehiepsi·13 Aug 2019·3 Comments

No autodiff on linalg.solve

bug
clemisch picture clemisch·9 Aug 2019·3 Comments

jaxlib build w/ cuda: File not found during compilation

bug build
clemisch picture clemisch·2 Jul 2019·7 Comments

xla in pmap fails (i.e. jit-of-pmap or lax.scan with collectives)

bug
jheek picture jheek·3 Jun 2019·5 Comments
👍1

2nd order derivs wants insane amount of ram

question
proteneer picture proteneer·30 May 2019·8 Comments

GPU determinism flag

documentation question
j-towns picture j-towns·3 Apr 2019·4 Comments
👍2

np.arange weirdness with jit.

bug
sschoenholz picture sschoenholz·31 Mar 2019·6 Comments

jit cache leaks memory

bug
jonasrauber picture jonasrauber·27 Jan 2019·4 Comments
👎1 👍1

question about np.random api

question
HerculesJack picture HerculesJack·18 Jan 2019·2 Comments

How to use jax.hessian?

enhancement question
zafarali picture zafarali·29 Dec 2018·4 Comments
❤1
 Previous issues