jax.lax.reduce\_window ====================== .. currentmodule:: jax.lax .. autofunction:: reduce_window