toma (pypi): Write algorithms in PyTorch that adapt to the available (CUDA) memory
Keywords: tools|pytorch|data-science|gpu|machine-learning|python