Top jax issues

jit decorator can't accept arguments

duvenaud picture duvenaud·1 Jan 2019·6 Comments

np.linalg.inv support

enhancement
holmrenser picture holmrenser·10 Dec 2018·9 Comments

jax.lax.switch host memory leak

performance
cooijmanstim picture cooijmanstim·16 Dec 2020·11 Comments

Add a "How to think in JAX" doc

documentation
jakevdp picture jakevdp·19 Oct 2020·5 Comments
👍14

Prevent custom calls with side effects to be optimized out

enhancement
dionhaefner picture dionhaefner·23 Jul 2020·22 Comments
👀1 👍1

Does vmap or lax.scan do caching?

question
ericmjl picture ericmjl·21 Jul 2020·6 Comments

Jaxify numpy function

question
sursu picture sursu·20 Apr 2020·3 Comments

JAX build on CentOS 7 (RHEL variant)

MilesCranmer picture MilesCranmer·27 Jan 2020·10 Comments

seg fault happens when running CPU code using GPU-supported jaxlib

bug
fehiepsi picture fehiepsi·23 Aug 2019·11 Comments

Clear GPU memory

question
clemisch picture clemisch·21 Aug 2019·11 Comments
👍2
 Previous issues
Next issues