jax.numpy.moveaxis ================== .. currentmodule:: jax.numpy .. autofunction:: moveaxis