jax.lax.pmean ============= .. currentmodule:: jax.lax .. autofunction:: pmean