Several simple examples for popular neural network toolkits calling custom CUDA operators.
☆1,540Apr 29, 2021Updated 5 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆2,800Jan 16, 2024Updated 2 years ago
- how to optimize some algorithm in cuda.☆3,277Updated this week
- 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,360Jul 29, 2023Updated 3 years ago
- compiler learning resources collect.☆2,777May 20, 2026Updated 4 months ago
- Sample codes for my CUDA programming book☆2,096Dec 14, 2025Updated 9 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Open sources book with Modern CUDA Learn Notes for Beginners, includes FP16/BF16, FP8, HGEMM, FlashAttention, CuTe, etc.☆11,983Updated this week
- A simple high performance CUDA GEMM implementation.☆440Jan 4, 2024Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆6,448Mar 27, 2024Updated 2 years ago
- learning how CUDA works☆407Mar 3, 2025Updated last year
- LightSeq: A High Performance Library for Sequence Processing and Generation☆3,294May 16, 2023Updated 3 years ago
- Material for gpu-mode lectures☆6,617Sep 9, 2026Updated last week
- Examples of CUDA implementations by Cutlass CuTe☆282Jul 1, 2025Updated last year
- Samples for CUDA Developers which demonstrates features in CUDA Toolkit☆9,643Sep 9, 2026Updated last week
- ☆1,051Mar 13, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆426Jan 2, 2025Updated last year
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆10,457Updated this week
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,822Mar 28, 2024Updated 2 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆535Jan 20, 2026Updated 8 months ago
- Simple samples for TensorRT programming☆1,673Sep 13, 2026Updated last week
- 校招、秋招、春招、实习好项目！带你从零实现一个高性能的深度学习推理库，支持大模型 llama2 、Unet、Yolov5、Resnet等模型的推理。Implement a high-performance deep learning inference library st…☆3,500Jun 22, 2025Updated last year
- Development repository for the Triton language and compiler☆20,197Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆13,354Sep 10, 2026Updated last week
- 《Machine Learning Systems: Design and Implementation》 (V2 is launching soon）☆4,860Mar 15, 2026Updated 6 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- FlashInfer: Kernel Library for LLM Serving☆6,452Updated this week
- Lightning fast C++/CUDA neural network framework☆4,536Apr 21, 2026Updated 4 months ago
- Learn CUDA Programming, published by Packt☆1,264Dec 30, 2023Updated 2 years ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆135Aug 12, 2023Updated 3 years ago
- A primitive library for neural network☆1,367Nov 24, 2024Updated last year
- 关于书籍CUDA Programming使用了pycuda模块的Python版本的示例代码☆260Jun 22, 2020Updated 6 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Jun 21, 2026Updated 2 months ago
- 📚A curated list of Awesome LLM/VLM Inference Papers with Codes: Flash-Attention, Paged-Attention, WINT8/4, Parallelism, etc.🎉☆5,508Aug 14, 2026Updated last month
- Implementation of popular deep learning networks with TensorRT network definition API☆7,833Aug 18, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- System for AI Education Resource.☆4,344Oct 25, 2024Updated last year
- AISystem 主要是指AI系统，包括AI芯片、AI编译器、AI推理和训练框架等AI全栈底层技术☆17,857Sep 3, 2025Updated last year
- CUDA Library Samples☆2,511Sep 11, 2026Updated last week
- Open Machine Learning Compiler Framework☆13,764Updated this week
- Yinghan's Code Sample☆367Jul 25, 2022Updated 4 years ago
- ☆192May 11, 2026Updated 4 months ago
- Fast and memory-efficient exact attention☆24,976Updated this week