Top jax issues

Trouble using JIT when the function is in a class?

question
vedantammihir picture vedantammihir·26 Aug 2019·5 Comments

Limit jax multithreading

enhancement
samuela picture samuela·21 May 2019·8 Comments

jit `lax.scatter_add` does not improve performance in CPU

enhancement
fehiepsi picture fehiepsi·10 May 2019·12 Comments

Any way to avoid recompiling of jit'ed sub-functions?

question
gnool picture gnool·14 Oct 2020·13 Comments
👍1

FFT precision/performance

performance
zymone picture zymone·4 May 2020·20 Comments
👍1

More flexible ODE integration

samuela picture samuela·7 Apr 2020·27 Comments
👍6 🎉1

Implement all numpy.fft operations

enhancement good first issue
shoyer picture shoyer·17 Dec 2019·15 Comments
👍1

Recurrent layer in stax?

enhancement
nafizh picture nafizh·26 Apr 2019·9 Comments

Building error on ARM Device

build
AashinShazar picture AashinShazar·21 Nov 2020·19 Comments
👍1

Using JAX with Pandas

enhancement
harshit-2115 picture harshit-2115·8 May 2020·3 Comments
 Previous issues
Next issues