MLX

einx.mlx.adapt_with_vmap	Adapts an operation to einx notation using mlx.core.vmap.
einx.mlx.adapt_numpylike_reduce	Adapts an operation with a Numpy-like reduce signature to einx notation.
einx.mlx.adapt_numpylike_elementwise	Adapts an operation with a Numpy-like element-wise signature to einx notation.