jax.lax.pad =========== .. currentmodule:: jax.lax .. autofunction:: pad