Top jax issues

Add graph_grad function

enhancement
IlyaOrson picture IlyaOrson·11 Dec 2020·6 Comments
👍3

CUDNN_STATUS_INTERNAL_ERROR

question
fmahdisoltani picture fmahdisoltani·12 Nov 2020·8 Comments

jax.numpy.linalg.matrix_power does not play well with jit

enhancement
martiningram picture martiningram·28 Oct 2020·7 Comments

Sparse-sparse matmul

question
ljmartin picture ljmartin·26 Oct 2020·2 Comments

Upgrading from 0.2.3 seems to break something with custom VJP

bug
NeilGirdhar picture NeilGirdhar·22 Oct 2020·23 Comments

Question: What's the best way to implement a ring buffer?

KristianHolsheimer picture KristianHolsheimer·15 Oct 2020·6 Comments

factor named_call primitive into jax core

enhancement
froystig picture froystig·13 Oct 2020·4 Comments

JAX Differentially Private SGD Example Compilation Very Slow

bug
n2cholas picture n2cholas·3 Oct 2020·3 Comments

Suggestion: expose `_jit_is_disabled` utility

enhancement
KristianHolsheimer picture KristianHolsheimer·8 Aug 2020·4 Comments

Cannot print a doubly-reshaped ShardedDeviceArray

bug
fehiepsi picture fehiepsi·29 Jun 2020·7 Comments
 Previous issues
Next issues