jax

80f963c0 - Fix mutable array effects not being tracked properly