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

This has been deprecated since JAX version 0.3.7, and will be removed in jax-ml/jax#11944. The jax.numpy.tile() function is a direct replacement, but in some places arrays can be constructed more efficiently using `lax.broadcasted_iota`.

PiperOrigin-RevId: 468227966

jax3d/math/volume_rendering.py

@@ -70,7 +70,7 @@ def sample_1d(
   # Initialize t with leading bin edges
   t = jnp.linspace(0.0, 1.0, sample_count, endpoint=False, dtype=dtype)
   t = t.reshape(*([1] * len(batch_shape)), sample_count)
-  t = t.tile(tuple(batch_shape) + (1,))
+  t = jnp.tile(t, tuple(batch_shape) + (1,))
 
   if strategy == SamplingStrategy.STRATIFIED:
     # Randomly perturb points within depth bins
@@ -106,7 +106,7 @@ def sample_1d_grid(
   t = jnp.linspace(0.0, 1.0, sample_count, endpoint=False, dtype=dtype)
   t += 0.5 / sample_count
   t = t.reshape(*([1] * len(batch_shape)), sample_count)
-  t = t.tile(tuple(batch_shape) + (1,))
+  t = jnp.tile(t, tuple(batch_shape) + (1,))
   return t