jax.numpy.stack =============== .. currentmodule:: jax.numpy .. autofunction:: stack