Top jax issues

Implementation of np.interp

enhancement question
AdrienCorenflos picture AdrienCorenflos·25 Jul 2020·19 Comments

custom_vjp can't handle vmapping over a nondiff_argnums arg

bug
mattjj picture mattjj·21 Jul 2020·5 Comments

JAX/TF2 on multi-GPU silently corrupted computations

bug question
david-berthelot picture david-berthelot·20 Jul 2020·12 Comments

Computing the diagonal elements of a Hessian

question
Marius1311 picture Marius1311·20 Jul 2020·9 Comments

Performance issue with sparse dot product using jit/GPU

performance question
romanodev picture romanodev·11 Jul 2020·2 Comments

ode is not working in jax 0.1.70

bug
fehiepsi picture fehiepsi·27 Jun 2020·7 Comments

DeviceArray's method .copy() returns np.array

question
Daulbaev picture Daulbaev·17 Jun 2020·6 Comments

Create `Primitive` with only impl and vjp

question
samuela picture samuela·12 Jun 2020·8 Comments

vmap(pmap) different results than map(pmap)

bug
btanner picture btanner·11 Jun 2020·2 Comments

Scaling behaviour of jax.test_util.check_grads

available bug
nsavinov picture nsavinov·18 May 2020·7 Comments
 Previous issues
Next issues