Top jax issues

vjp of dtype promotion silently truncates complex numbers

question useful_read
shoyer picture shoyer·11 Jun 2020·12 Comments

Incorrect output from jax.numpy.column_stack

bug
spenrich picture spenrich·4 Jun 2020·2 Comments

vmap is incompatible with custom pytree

question
fehiepsi picture fehiepsi·31 May 2020·6 Comments

Unexpectedly high grad-of-scan memory usage

question
jeffgortmaker picture jeffgortmaker·22 May 2020·4 Comments

Tracing a function that indexes into Numpy array gives a poor error message

better_errors
john-m-jumper picture john-m-jumper·18 May 2020·4 Comments

polyval is exceedingly slow to compile for high degree polynomials

performance
shoyer picture shoyer·12 May 2020·8 Comments
👍1

Forward-mode differentiation rule for 'custom_lin' not implemented

bug
jacobjinkelly picture jacobjinkelly·21 Apr 2020·2 Comments

Support multinode training on GPU

enhancement
py4 picture py4·16 Apr 2020·9 Comments

TypeError: <class 'jax.ad_util.Zero'> is not a valid Jax type when I combine custom_jvp and lax.scan

bug
rdaems picture rdaems·9 Apr 2020·4 Comments

Design suggestion: allow objects to be closed under JAX's transformation rules

NeilGirdhar picture NeilGirdhar·28 Feb 2020·17 Comments
 Previous issues
Next issues