Top jax issues

Jax script killed by OS when evaluating grad() of complex function for first time - only with GPU version

Victorlouisdg picture Victorlouisdg·23 Dec 2019·14 Comments

Numba bridge

shoyer picture shoyer·16 Dec 2019·12 Comments
👍6

Overrides of NumPy functions on JAX arrays

enhancement
shoyer picture shoyer·24 Oct 2019·6 Comments
👍4

Compatibility with original numpy and scipy

lonelykid picture lonelykid·15 Oct 2019·3 Comments
👍3

Allowing pmap on CPUs by making CPU cores visibles directly on Jax.

sguada picture sguada·27 Sep 2019·12 Comments
👍12

No vmap over multiple axes for multiple inputs

clemisch picture clemisch·12 Aug 2019·4 Comments

Class inheriting nametuple is not treated as a valid JAX Type

zhongwen picture zhongwen·3 Jun 2019·3 Comments

backward pass of scan is very slow to compile in CPU

bug
fehiepsi picture fehiepsi·26 May 2019·10 Comments

Pickling a jitted function

application enhancement
adamhaber picture adamhaber·6 May 2019·16 Comments

lax.cond does not exist, contrary to the README

jonasrauber picture jonasrauber·5 Feb 2019·5 Comments
 Previous issues
Next issues