jax.numpy.array\_split ====================== .. currentmodule:: jax.numpy .. autofunction:: array_split