code: def jax_funcify_BatchedDot(op, **kwargs):
      def batched_dot(a, b):
          if a.shape[0] != b.shape[0]:
              raise TypeError("Shapes must match in the 0-th dimension")
          return jnp.matmul(a, b)
  
      return batched_dot
  
file_path: pytensor/pytensor/link/jax/dispatch/nlinalg.py
imports: import jax.numpy
  from pytensor.link.jax.dispatch import jax_funcify
  from pytensor.tensor.blas import BatchedDot
  from pytensor.tensor.math import Dot, MaxAndArgmax
  from pytensor.tensor.nlinalg import SVD, Det, Eig, Eigh, MatrixInverse, MatrixPinv, QRFull, SLogDet
label_desc: Use image node_id 10 for calling a global function with example usage: jax_funcify_BatchedDot(op) and returns: batched_dot, jnp