Avoid using deprecated jax.numpy.ndarray.tile() method #249

Closed aterzis-google closed 2 years ago