jax

84210a32 - jnp.repeat: don't cast repeats to array, as they must be static.