jax.lax.sub =========== .. currentmodule:: jax.lax .. autofunction:: sub