Fix repeat.jax
close #98