Top jax issues

Multiple Workers with PyTorch Dataloader

n2cholas picture n2cholas·9 Jun 2020·6 Comments
❤1

Please consider facilitating debugging leaked tracers

NeilGirdhar picture NeilGirdhar·13 May 2020·2 Comments

Inverse Fisher/Hessian as GD Preconditioner

shannon63 picture shannon63·31 Mar 2020·3 Comments

np.unwrap runtime explodes

clemisch picture clemisch·15 Mar 2020·7 Comments

jax.jacobian could automatically choose between forward and reverse mode

shoyer picture shoyer·2 Mar 2020·2 Comments
👍1

Add multivariate normal pdf evalutation to jax.scipy

enhancement
RobertTLange picture RobertTLange·26 Feb 2020·3 Comments

Einsum is slow

romanngg picture romanngg·4 Feb 2020·5 Comments

cholesky not working with vmap on complex arrays

txie-93 picture txie-93·6 Jan 2020·9 Comments

Compiling JAX from inside a Dockerfile fails

adityaiitb picture adityaiitb·6 Sep 2019·4 Comments

pip install problem

build contributions welcome
BEpresent picture BEpresent·6 Aug 2019·12 Comments
 Previous issues
Next issues