Add einsum dispatch for mlx#2017

jessegrabowski wants to merge 2 commits intopymc-devs:v3pymc-devs/pytensor:v3from

jessegrabowski:mlx-einsumjessegrabowski/pytensor:mlx-einsumCopy head branch name to clipboard