FlashComm EP Kernels - Elegant Python host-side binding for LittleKernel. This module mirrors FlashComm's EPKernels class but uses LittleKernel's build() API instead of C++ extension modules. All CUDA ...
This project implements a complete Transformer inference engine using only hand-written CUDA kernels. Every GPU operation — matrix multiplication, attention, normalization, activation functions — is ...