PyTorch
-------

..  autosummary::
    :toctree: torch
    :nosignatures:

    einx.torch.adapt_with_vmap
    einx.torch.adapt_numpylike_reduce
    einx.torch.adapt_numpylike_elementwise