[mgunyho]

For Torch, I have come across Named Tensors, which should work in a similar way: https://docs.pytorch.org/docs/stable/named_tensor.html

The docs say that it's a prototype feature, and I think it has been that way for a few years now, so no idea how production-ready it is.

[xiphias2]

It's a much worse API than Xarrays, it seems like somebody should build it on top of PyTorch.