jax.numpy.apply\_along\_axis ============================ .. currentmodule:: jax.numpy .. autofunction:: apply_along_axis