GitHub repository: nicholas-leonard/cunnx
Primary language: Cuda
Description: Experimental cuda nn package