PyTorchModelMemoryProfiler Module

Helpers to profile PyTorch model memory usage during training and inference.