jax.lax.batch\_matmul
=====================

.. currentmodule:: jax.lax

.. autofunction:: batch_matmul