jax.lax.full ============ .. currentmodule:: jax.lax .. autofunction:: full