luliyucoordinate / flash-attention-minimal
Flash Attention in ~100 lines of CUDA (forward pass only)
☆10Updated 9 months ago
Alternatives and similar repositories for flash-attention-minimal:
Users that are interested in flash-attention-minimal are comparing it to the libraries listed below
- Awesome code, projects, books, etc. related to CUDA☆16Updated this week
- ☆24Updated last year
- 使用 CUDA C++ 实现的 llama 模型推理框架☆48Updated 4 months ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated last year
- TensorRT encapsulation, learn, rewrite, practice.☆28Updated 2 years ago
- ffmpeg+cuvid+tensorrt+multicamera☆12Updated 3 months ago
- Optimize softmax in triton in many cases☆20Updated 6 months ago
- This is a repository to practice multi-thread programming in C++☆22Updated last year
- 搜藏的希望的代码片段☆13Updated last year
- Llama3 Streaming Chat Sample☆22Updated 11 months ago
- A light llama-like llm inference framework based on the triton kernel.☆100Updated 3 weeks ago
- async inference for machine learning model☆26Updated 2 years ago
- Multiple GEMM operators are constructed with cutlass to support LLM inference.☆17Updated 6 months ago
- NVIDIA TensorRT Hackathon 2023复赛选题:通义千问Qwen-7B用TensorRT-LLM模型搭建及优化☆41Updated last year
- For 2022 Nvidia Hackathon☆20Updated 2 years ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆46Updated last year
- ☆19Updated last year
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆35Updated 3 weeks ago
- <Good Luck To You!> 's work for <TRT-Hackathon-2022-final>☆7Updated 2 years ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆60Updated 7 months ago
- ☆29Updated 11 months ago
- CPU Memory Compiler and Parallel programing☆25Updated 4 months ago
- learn TensorRT from scratch🥰☆13Updated 6 months ago
- bilibili视频【CUDA 12.x 并行编程入门(C++版)】配套代码☆30Updated 7 months ago
- ☆10Updated 3 weeks ago
- ☆48Updated 2 months ago
- Created a simple neural network using C++17 standard and the Eigen library that supports both forward and backward propagation.☆9Updated 8 months ago
- An onnx-based quantitation tool.☆71Updated last year
- EasyNN是一个面向教学而开发的神经网络推理框架,旨在让大家0基础也能自主完成推理框架编写!☆26Updated 7 months ago
- ☆15Updated last year