The MTIA backend is implemented out of the tree, only interfaces are be defined here.
.. automodule:: torch.mtia.memory
.. currentmodule:: torch.mtia.memory
.. autosummary:: :toctree: generated :nosignatures: memory_stats