Usage
=====
.. automodule:: torch_max_mem.api
    :members: