jax.lax.dot =========== .. currentmodule:: jax.lax .. autofunction:: dot