LLM training in simple, raw C/CUDA

DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling

cuGraph - RAPIDS Graph Analytics Library

GPU Accelerated t-SNE for CUDA with Python bindings

RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing …

A shift-window based transformer for 3D sparse tasks

Implementation of fused cosine similarity attention in the same style as Flash Attention

this is a high performance cuda porting of cbow model of word2vec

cuSTSG is a GPU-enabled spatial-temporal Savitzky-Golay (STSG) program based on the Compute Unified Device Architecture (CUDA). Firstly, the cosine similarities between the annual NDVI time series …