godweiyang / NN-CUDA-Example
Several simple examples for popular neural network toolkits calling custom CUDA operators.
☆1,389Updated 3 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
- how to optimize some algorithm in cuda.☆1,827Updated this week
- ☆1,008Updated 10 months ago
- Simple samples for TensorRT programming☆1,565Updated last month
- ☆2,279Updated last year
- PyTorch Project Specification.☆672Updated 3 years ago
- Sample codes for my CUDA programming book☆1,617Updated last year
- Parallel programming tutorials☆611Updated 3 years ago
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,610Updated 9 months ago
- A quickstart and benchmark for pytorch distributed training.☆1,650Updated 5 months ago
- compiler learning resources collect.☆2,238Updated 7 months ago
- ☆598Updated 7 months ago
- Some tricks of pytorch...☆1,177Updated 6 months 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…☆889Updated last year
- how to learn PyTorch and OneFlow☆378Updated 9 months ago
- 📚200+ Tensor/CUDA Cores Kernels, ⚡️flash-attn-mma, ⚡️hgemm with WMMA, MMA and CuTe (98%~100% TFLOPS of cuBLAS/FA2 🎉🎉).☆1,979Updated this week
- pytorch memory track code☆1,012Updated 3 years ago
- This is a Chinese translation of the CUDA programming guide☆1,376Updated 2 months ago
- real Transformer TeraFLOPS on various GPUs☆892Updated last year
- 整理 pytorch 单机多 GPU 训练方法与原理☆791Updated 3 years ago
- A simple deep learning framework in pure python for purpose of learning in DL☆431Updated 2 years ago
- row-major matmul optimization