Top jax issues

Optix documentation

joaogui1 picture joaogui1·24 Feb 2020·5 Comments

Using grad on function using lax.cond raises unclear assertion error

question
rdaems picture rdaems·18 Feb 2020·3 Comments

Should ShapeDtypeStruct.size exist?

enhancement
girving picture girving·6 Feb 2020·5 Comments

Matrix multiplication precision API

shoyer picture shoyer·4 Feb 2020·6 Comments
👍7

Support string arrays

question
romanngg picture romanngg·27 Jan 2020·2 Comments

Add jax.random.choice

enhancement
Thenerdstation picture Thenerdstation·24 Jan 2020·12 Comments

jax.random.randint silently returns zero for large values of maxval

better_errors good first issue
vmikulik picture vmikulik·15 Jan 2020·10 Comments

Could we have support for scipy.stats.logistic?

enhancement good first issue
rhalbersma picture rhalbersma·12 Jan 2020·4 Comments

slogdet timing differences

bug
reziproke87 picture reziproke87·23 Dec 2019·6 Comments

jax.lax.scan benefits apart from compilation time

documentation
clemisch picture clemisch·19 Dec 2019·2 Comments
 Previous issues
Next issues