/Custom_PyTorch_Cpp_and_CUDA_Modules

Custom C++ & CUDA modules for PyTorch (Mainly LSTM variants)

Primary LanguageJupyter Notebook

Stargazers