You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The jax.numpy.insert() function returns an incorrect result (zero-padding the array) when compiled with jax.jit. When not jitted, the results are correct.
Description
(pasting from Apple Developer forum)
The
jax.numpy.insert()
function returns an incorrect result (zero-padding the array) when compiled withjax.jit
. When notjit
ted, the results are correct.MWE
Output
System info (python version, jaxlib version, accelerator, etc.)
The text was updated successfully, but these errors were encountered: