GitHub repository: mberr/torch-max-mem
Primary language: Python
Description: Decorators for maximizing memory utilization with PyTorch & CUDA