matrix97317 / OneNeuralNetwork
This is a cross-chip platform collection of operators and a unified neural network library.
☆17Updated last year
Alternatives and similar repositories for OneNeuralNetwork:
Users that are interested in OneNeuralNetwork are comparing it to the libraries listed below
- ☆88Updated 3 weeks ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆181Updated 2 months ago
- Examples of CUDA implementations by Cutlass CuTe☆156Updated 2 months ago
- hands on model tuning with TVM and profile it on a Mac M1, x86 CPU, and GTX-1080 GPU.☆47Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆70Updated 3 weeks ago
- 分层解耦的深度学习推理引擎☆72Updated 2 months ago
- play gemm with tvm☆90Updated last year
- Tutorials for writing high-performance GPU operators in AI frameworks.☆130Updated last year
- ☆148Updated 3 months ago
- A tutorial for CUDA&PyTorch☆132Updated 3 months ago
- ☆109Updated last year
- ☆14Updated 3 years ago
- ☆60Updated this week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆63Updated 8 months ago
- ☆20Updated 4 years ago
- Machine Learning Compiler Road Map☆43Updated last year
- ☆96Updated 3 years ago
- ☆122Updated last year
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆82Updated 2 years ago
- ☆138Updated 3 months ago
- Triton Compiler related materials.☆28Updated 3 months ago
- CUDA 6大并行计算模式 代码与笔记☆60Updated 4 years ago
- ☆70Updated 2 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆91Updated 3 weeks ago
- 使用 cutlass 实现 flash-attention 精简版,具有教学意义☆39Updated 8 months ago
- ☆61Updated 3 months ago
- Free resource for the book AI Compiler Development Guide☆43Updated 2 years ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆35Updated last month
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆119Updated 2 years ago
- ☆92Updated 7 months ago