godweiyang / NN-CUDA-Example
Several simple examples for popular neural network toolkits calling custom CUDA operators.
☆1,339Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for NN-CUDA-Example
- how to optimize some algorithm in cuda.☆1,593Updated last week
- ☆990Updated 8 months ago
- Simple samples for TensorRT programming☆1,519Updated 2 weeks ago
- PyTorch Project Specification.☆667Updated 3 years ago
- ☆589Updated 5 months ago
- PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.☆1,558Updated 7 months ago
- ☆2,205Updated 10 months ago
- 📚Modern CUDA Learn Notes with PyTorch: Tensor/CUDA Cores, 📖150+ CUDA Kernels with PyTorch bindings, 📖HGEMM/SGEMM (95%~99% cuBLAS perfo…☆1,473Updated this week
- A quickstart and benchmark for pytorch distributed training.☆1,640Updated 3 months ago
- Some tricks of pytorch...☆1,160Updated 5 months ago
- Sample codes for my CUDA programming book☆1,572Updated last year
- pytorch memory track code☆1,003Updated 3 years 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…☆827Updated last year
- compiler learning resources collect.☆2,158Updated 5 months ago
- Parallel programming tutorials☆603Updated 3 years ago
- Model Quantization Benchmark☆765Updated 5 months ago
- This is a Chinese translation of the CUDA programming guide☆1,282Updated last week
- how to learn PyTorch and OneFlow☆348Updated 7 months ago
- real Transformer TeraFLOPS on various GPUs☆876Updated 10 months ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆589Updated 4 years ago
- A self-learning tutorail for CUDA High Performance Programing.☆257Updated last week
- 整理 pytorch 单机多 GPU 训练方法与原理☆775Updated 2 years ago
- A simple deep learning framework in pure python for purpose of learning in DL☆428Updated 2 years ago
- ☆228Updated 2 years ago
- row-major matmul optimization☆591Updated last year
- The road to hack SysML and become an system expert☆437Updated last month
- An easy/swift-to-adapt PyTorch-Lighting template. 套壳模板,简单易用,稍改原来Pytorch代码,即可适配Lightning。You can translate your previous Pytorch code much…☆1,346Updated last year
- MegCC是一个运行时超轻量,高效,移植简单的深度学习模型编译器☆473Updated 3 weeks ago
- A primitive library for neural network☆1,295Updated 2 weeks ago
- ppl.cv is a high-performance image processing library of openPPL supporting various platforms.☆493Updated 3 weeks ago