jax

1ac20854 - Fix "unhashable type" error when passing a jax array as the "repeats" argument to jnp.repeat().