Several simple examples for popular neural network toolkits calling custom CUDA operators.
☆1,526Apr 29, 2021Updated 4 years ago
Alternatives and similar repositories for NN-CUDA-Example
Users that are interested in NN-CUDA-Example are comparing it to the libraries listed below
Sorting:
- ☆2,698Jan 16, 2024Updated 2 years ago
- how to optimize some algorithm in cuda.☆2,825Feb 15, 2026Updated 2 weeks ago
- This is a series of GPU optimization topics. Here we will introduce how to optimize the CUDA kernel in detail. I will introduce several…☆1,244Jul 29, 2023Updated 2 years ago
- Sample codes for my CUDA programming book☆2,010Dec 14, 2025Updated 2 months ago
- 📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.🎉☆9,755Updated this week
- compiler learning resources collect.☆2,684Mar 19, 2025Updated 11 months ago
- A simple high performance CUDA GEMM implementation.☆426Jan 4, 2024Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆6,394Mar 27, 2024Updated last year
- learning how CUDA works☆376Mar 3, 2025Updated 11 months ago
- LightSeq: A High Performance Library for Sequence Processing and Generation☆3,303May 16, 2023Updated 2 years ago
- Material for gpu-mode lectures☆5,773Feb 1, 2026Updated last month
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,315Updated this week
- flash attention tutorial written in python, triton, cuda, cutlass☆488Jan 20, 2026Updated last month
- ☆1,047Mar 13, 2024Updated last year
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆8,870Jan 6, 2026Updated last month
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,785Mar 28, 2024Updated last year
- Simple samples for TensorRT programming☆1,658Jan 22, 2026Updated last month
- 校招、秋招、春招、实习好项目！带你从零实现一个高性能的深度学习推理库，支持大模型 llama2 、Unet、Yolov5、Resnet等模型的推理。Implement a high-performance deep learning inference library st…☆3,324Jun 22, 2025Updated 8 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,723Updated this week
- Lightning fast C++/CUDA neural network framework☆4,418Dec 14, 2025Updated 2 months ago
- A primitive library for neural network☆1,366Nov 24, 2024Updated last year
- 《Machine Learning Systems: Design and Implementation》- Chinese Version☆4,764Apr 13, 2024Updated last year
- Learn CUDA Programming, published by Packt☆1,231Dec 30, 2023Updated 2 years ago
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆407Jan 2, 2025Updated last year
- Development repository for the Triton language and compiler☆18,501Updated this week
- Examples of CUDA implementations by Cutlass CuTe☆269Jul 1, 2025Updated 8 months ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆135Aug 12, 2023Updated 2 years ago
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,652Jan 22, 2026Updated last month
- Implementation of popular deep learning networks with TensorRT network definition API☆7,680Updated this week
- FlashInfer: Kernel Library for LLM Serving☆5,009Feb 23, 2026Updated last week
- how to learn PyTorch and OneFlow☆485Mar 22, 2024Updated last year
- 📚A curated list of Awesome LLM/VLM Inference Papers with Codes: Flash-Attention, Paged-Attention, WINT8/4, Parallelism, etc.🎉☆5,022Updated this week
- System for AI Education Resource.☆4,222Oct 25, 2024Updated last year
- AISystem 主要是指AI系统，包括AI芯片、AI编译器、AI推理和训练框架等AI全栈底层技术☆16,340Sep 3, 2025Updated 5 months ago
- CUDA Library Samples☆2,324Feb 21, 2026Updated last week
- Fast and memory-efficient exact attention☆22,361Updated this week
- ONNX-TensorRT: TensorRT backend for ONNX☆3,188Feb 3, 2026Updated 3 weeks ago
- Open Machine Learning Compiler Framework☆13,142Updated this week
- Ongoing research training transformer models at scale☆15,461Updated this week