Log GPU memory usage at a given training stage.
(stage: str, device=None, rank=0)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected