jax.lax.scaled\_dot
===================

.. currentmodule:: jax.lax

.. autofunction:: scaled_dot