MLX
---

..  autosummary::
    :toctree: mlx
    :nosignatures:

    einx.mlx.adapt_with_vmap
    einx.mlx.adapt_numpylike_reduce
    einx.mlx.adapt_numpylike_elementwise