Koila

github.com ↗

A simple wrapper around pytorch that prevents CUDA out of memory issues.