jax.nn.scaled\_matmul
=====================

.. currentmodule:: jax.nn

.. autofunction:: scaled_matmul