jax

48e14dcc - Implement mutation by replacing the contents of a jax.Array with a result