luliyucoordinate / flash-attention-minimalLinks
Flash Attention in ~100 lines of CUDA (forward pass only)
☆11Updated 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☆28Updated last month
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated 2 years ago
- 记录量化LLM中的总结。☆55Updated last week
- A light llama-like llm inference framework based on the triton kernel.☆168Updated last week
- 使用 CUDA C++ 实现的 llama 模型推理框架☆63Updated last year
- 一个轻量化 的大模型推理框架☆20Updated 7 months ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆51Updated 2 years ago
- ☆39Updated 8 months ago
- ☆26Updated 2 years ago
- ☆14Updated 2 months ago
- SGEMM optimization with cuda step by step☆21Updated last year
- 🎉My Collections of CUDA Kernels~☆11Updated last year
- This is a repository to practice multi-thread programming in C++☆26Updated last year
- 🐱 ncnn int8 模型量化评估☆14Updated 3 years ago
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆25Updated this week
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆78Updated last year
- ☆17Updated 2 years ago
- Inference deployment of the llama3☆11Updated last year
- ☆49Updated last year
- Implement Flash Attention using Cute.☆100Updated last year
- CUDA 6大并行计算模式 代码与笔记☆61Updated 5 years ago
- An onnx-based quantitation tool.☆71Updated 2 years ago
- ☆21Updated 4 years ago
- Llama3 Streaming Chat Sample☆22Updated last year
- DETR tensor去除推理过程无用辅助头+fp16部署再次加速+解决转tensorrt 输出全为0问题的新方法。☆12Updated 2 years ago
- Optimize softmax in triton in many cases☆21Updated last year
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆20Updated 5 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆44Updated 10 months ago
- 搜藏的希望的代码片段☆13Updated 2 years ago
- CPU Memory Compiler and Parallel programing☆26Updated last year