Top jax issues

gradients through np.where when one of branches is nan, #347 except with arctan2

question
cpgoodri picture cpgoodri·23 Jul 2019·7 Comments

what's a good way to save jax arrays?

enhancement question
rowanz picture rowanz·14 May 2019·5 Comments

Support ahead of time compilation

enhancement
apark263 picture apark263·4 Mar 2019·2 Comments
👍12

add examples and docs for how to use lax.conv functions

enhancement good first issue
mattjj picture mattjj·11 Jan 2019·9 Comments

Feature request: export TF ops

enhancement
kovasb picture kovasb·10 Dec 2018·10 Comments

Not able to calculate logpdf using jax.scipy.stats.multivariate_normal

nitinkmittal picture nitinkmittal·17 Nov 2020·3 Comments

any suggestions on how to improve performance for gradient step with odeint?

ibulu picture ibulu·8 Aug 2020·29 Comments

TPU not detected by jax in Colab

Tenoke picture Tenoke·24 Jan 2020·20 Comments

float128 please??

enhancement
mc2engel picture mc2engel·15 Jan 2020·8 Comments

jax tensor and numpy array convertion

documentation question
yfji picture yfji·8 Jan 2020·3 Comments
👍3
 Previous issues
Next issues