jax.lax.conv ============ .. currentmodule:: jax.lax .. autofunction:: conv