torch-max-mem (pypi): Maximize memory utilization with PyTorch.
Keywords: snekpack|cookiecutter|torch|cuda|python|pytorch