cutex (pypi): PyCUDA based PyTorch Extension Made Easy
Keywords: pycuda|pytorch