jax.lax.batch_matmul# jax.lax.batch_matmul(lhs, rhs, precision=None)[source]# Batch matrix multiplication. Parameters: lhs (Array) rhs (Array) precision (str | Precision | tuple[str, str] | tuple[Precision, Precision] | None) Return type: Array