dhcode-cpp / online-softmaxLinks
simplest online-softmax notebook for explain Flash Attention
☆13Updated 11 months ago
Alternatives and similar repositories for online-softmax
Users that are interested in online-softmax are comparing it to the libraries listed below
Sorting:
- Pipeline-Parallel Lecture: Simplest Dualpipe Implementation.☆29Updated 3 months ago
- Implementation of FlashAttention in PyTorch☆176Updated 11 months ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆51Updated 2 years ago
- qwen-nsa☆85Updated 2 months ago
- ☆79Updated 2 years ago
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆131Updated last year
- ☆51Updated 2 years ago
- export llama to onnx☆137Updated 11 months ago
- Inference code for LLaMA models☆128Updated 2 years ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆110Updated 8 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆137Updated last month
- Transformer related optimization, including BERT, GPT☆59Updated 2 years ago
- ☆149Updated 5 months ago
- Transformer related optimization, including BERT, GPT☆17Updated 2 years ago
- DeepSeek Native Sparse Attention pytorch implementation☆109Updated last month
- 模型压缩的小白入门教程☆22Updated last year
- ☆66Updated 2 weeks ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated 2 years ago
- ☆154Updated 9 months ago
- Efficient, Flexible, and Highly Fault-Tolerant Model Service Management Based on SGLang☆61Updated last year
- 青稞Talk☆175Updated last week
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆70Updated 2 years ago
- Models and examples built with OneFlow☆100Updated last year
- Model compression toolkit engineered for enhanced usability, comprehensiveness, and efficiency.☆222Updated last week
- Triton Documentation in Chinese Simplified / Triton 中文文档☆95Updated this week
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆48Updated 4 months ago
- ☆37Updated 2 years ago
- DashInfer is a native LLM inference engine aiming to deliver industry-leading performance atop various hardware architectures, including …☆270Updated 4 months ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆62Updated last year
- Tutorials for writing high-performance GPU operators in AI frameworks.☆133Updated 2 years ago