luliyucoordinate / flash-attention-minimalLinks
Flash Attention in ~100 lines of CUDA (forward pass only)
☆10Updated last year
Alternatives and similar repositories for flash-attention-minimal
Users that are interested in flash-attention-minimal are comparing it to the libraries listed below
Sorting:
- Awesome code, projects, books, etc. related to CUDA☆23Updated 2 weeks ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated 2 years ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆60Updated 9 months ago
- 🐱 ncnn int8 模型量化评估☆13Updated 2 years ago
- 🎉My Collections of CUDA Kernels~☆11Updated last year
- 一个轻量化的大模型推理 框架☆20Updated 3 months ago
- This is a repository to practice multi-thread programming in C++☆25Updated last year
- An onnx-based quantitation tool.☆71Updated last year
- ☆26Updated 2 years ago
- A light llama-like llm inference framework based on the triton kernel.☆148Updated 3 weeks ago
- ffmpeg+cuvid+tensorrt+multicamera☆12Updated 8 months ago
- Llama3 Streaming Chat Sample☆22Updated last year
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆50Updated last year
- 搜藏的希望的代码片段☆13Updated 2 years ago
- DETR tensor去除推理过程无用辅助头+fp16部署再次加速+解决转tensorrt 输出全为0问题的新方法。☆12Updated last year
- EasyNN是一个面向教学而开发的神经网络推理框架,旨在让大家0基础也能自主完成推理框架编写!☆32Updated last year
- TensorRT encapsulation, learn, rewrite, practice.☆28Updated 2 years ago
- bilibili视频【CUDA 12.x 并行编程入门(C++版)】配套代码☆29Updated last year
- NVIDIA TensorRT Hackathon 2023复赛选题:通义千问Qwen-7B用TensorRT-LLM模型搭建及优化☆42Updated last year
- ☆35Updated 3 months ago
- For 2022 Nvidia Hackathon☆22Updated 3 years ago
- 该代码与B站上的视频 https://www.bilibili.com/video/BV18L41197Uz/?spm_id_from=333.788&vd_source=eefa4b6e337f16d87d87c2c357db8ca7 相关联。☆70Updated last year
- ☆17Updated last year
- ☆12Updated 5 months ago
- ☆30Updated 9 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆74Updated last year
- Quick and Self-Contained TensorRT Custom Plugin Implementation and Integration☆69Updated 3 months ago
- Quantize yolov7 using pytorch_quantization.🚀🚀🚀☆10Updated last year
- HunyuanDiT with TensorRT and libtorch☆17Updated last year
- SGEMM optimization with cuda step by step☆20Updated last year