torch-max-mem icon indicating copy to clipboard operation
torch-max-mem copied to clipboard

Decorators for maximizing memory utilization with PyTorch & CUDA

Results 5 torch-max-mem issues
Sort by recently updated
recently updated
newest added

https://github.com/mberr/torch-max-mem/actions/runs/15103471625/job/42448022137 ``` FileNotFoundError: [Errno 2] No such file or directory: 'uvx' ```

the improvements to pykeen in https://github.com/pykeen/pykeen/pull/975 don't work now that we have totally externalized AMO to this package - can we think about how to make sure that the CPU...

This is an autogenerated PR. Use this to reject the changes in this repository. [Cruft](https://cruft.github.io/cruft/) has detected updates from the Cookiecutter repository.

This is an autogenerated PR. Use this to merge the changes to this repository. [Cruft](https://cruft.github.io/cruft/) has detected updates from the Cookiecutter repository.