AI product Open source · MIT

DeepGEMM

DeepGEMM is a unified CUDA library of GPU tensor-core kernels for computation primitives used by large language models. It provides FP8, FP4, and BF16 GEMMs; grouped and masked GEMMs for mixture-of-experts models; fused Mega MoE kernels that overlap expert-parallel dispatch and combine with linear layers, SwiGLU, NVLink communication, and tensor-core computation; and non-paged and paged MQA scoring kernels for the DeepSeek v3.2 lightning indexer. The repository also includes weight-gradient kernels and utilities for tensor layouts, scaling factors, programmatic dependent launch, deterministic algorithms, and JIT configuration.

View repository