jax.lax.slice\_in\_dim ====================== .. currentmodule:: jax.lax .. autofunction:: slice_in_dim