Call MLX linalg ops directly on batched inputs instead of mx.vmap - #2425

Call MLX linalg ops directly on batched inputs instead of mx.vmap#2425

jessegrabowski merged 4 commits into

pymc-devs:mainpymc-devs/pytensor:mainfrom

jessegrabowski:mlx-natively-batched-linalgjessegrabowski/pytensor:mlx-natively-batched-linalgCopy head branch name to clipboard