Zilong He's repositories
LinearList
A faster implementation for Pytorch ModuleList of Linear Layers using custom C++ and CUDA extensions
Language:C++000
000
Language:CSS000
Sun Yat-sen University
Guangzhou, Guangdong
A faster implementation for Pytorch ModuleList of Linear Layers using custom C++ and CUDA extensions