Another push small

Author: cetagostini

pytensor/link/mlx/dispatch/elemwise.py

@@ -39,7 +39,7 @@ def prod(x):
     elif isinstance(op.scalar_op, AND):
 
         def all(x):
-            return mx.all(x, axis=op.axis)
+            return mx.all(a=x, axis=op.axis)
 
         return all
     elif isinstance(op.scalar_op, OR):

pytensor/link/mlx/dispatch/math.py

@@ -199,7 +199,7 @@ def neg(x):
     elif isinstance(op.scalar_op, AND):
 
         def all(x):
-            return mx.all(x)
+            return mx.all(a=x, axis=op.axis)
 
         return all
     elif isinstance(op.scalar_op, OR):