Top jax issues

Question about block_until_ready() on tuple

question
jacobjinkelly picture jacobjinkelly·17 May 2020·2 Comments

matmul slow for complex dtypes

bug performance
mganahl picture mganahl·12 May 2020·7 Comments

NaN in gradient of `np.linalg.norm` at 0

question
Victorlouisdg picture Victorlouisdg·12 May 2020·7 Comments

Questions on the implementation of reverse AD in jax

documentation question
refraction-ray picture refraction-ray·11 May 2020·7 Comments
👍1

jaxlib build error in PacketMathAVX512.h

bug build
clemisch picture clemisch·5 May 2020·3 Comments

stax.serial.apply_fun is not a valid JAX type inside odeint

better_errors documentation question
skrsna picture skrsna·1 May 2020·9 Comments

"logit" parameter of jax.random.categorical is misnamed, actually a log probability

question
grisaitis picture grisaitis·29 Apr 2020·7 Comments

apply constraints in optimizer

question
PgLoLo picture PgLoLo·25 Apr 2020·7 Comments

how to vectorize custom functions with vmap

question
cagrikymk picture cagrikymk·24 Mar 2020·7 Comments

cumtom_jvp fails when the function is real-valued and its gradient is complex-valued

bug
NeilGirdhar picture NeilGirdhar·23 Mar 2020·12 Comments
 Previous issues
Next issues