jax.lax.mul =========== .. currentmodule:: jax.lax .. autofunction:: mul