jax.numpy.apply\_over\_axes =========================== .. currentmodule:: jax.numpy .. autofunction:: apply_over_axes