Top jax issues

Using TPUs in coreless mode like with tf.device(None)

question
Tenoke picture Tenoke·29 Jan 2020·8 Comments

memory leak of jax.random in GPU

fehiepsi picture fehiepsi·18 Nov 2019·16 Comments

Cloud TPU Support

enhancement
aidangomez picture aidangomez·8 Dec 2018·6 Comments
👍1

Does lax cond short circuit?

contributions welcome documentation question
john-heyer picture john-heyer·15 May 2020·10 Comments

autodiff support for jax.numpy.linalg.eig

enhancement
froystig picture froystig·17 Apr 2020·34 Comments

JAX performance can be better (comparison with PyTorch)

sergei-bondarenko picture sergei-bondarenko·8 Dec 2019·7 Comments
👍1

PyPI macOS binaries missing for jaxlib 0.1.26

bug build
JuliusKunze picture JuliusKunze·27 Aug 2019·23 Comments

Add ability to place computation on a specific GPU

enhancement
afrozenator picture afrozenator·1 Jul 2019·15 Comments
👍3

Regression in performance of samplers

bug
fehiepsi picture fehiepsi·10 May 2019·7 Comments

Inconsistent batching behavior between MLPs and convnets

enhancement question
ericjang picture ericjang·15 Feb 2019·13 Comments
 Previous issues
Next issues