dhcode-cpp / online-softmaxLinks
simplest online-softmax notebook for explain Flash Attention
☆13Updated 10 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.☆27Updated last month
- Implementation of FlashAttention in PyTorch☆172Updated 9 months ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆71Updated 2 years ago
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆126Updated last year
- qwen-nsa☆83Updated 3 weeks ago
- 青稞Talk☆157Updated 2 weeks ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆50Updated 2 years ago
- from MHA, MQA, GQA to MLA by 苏剑林, with code☆33Updated 8 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆107Updated 7 months ago
- Inference code for LLaMA models☆127Updated 2 years ago
- ☆148Updated 4 months ago
- ☆51Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆17Updated 2 years ago
- Model compression toolkit engineered for enhanced usability, comprehensiveness, and efficiency.☆191Updated last week
- Efficient Mixture of Experts for LLM Paper List☆142Updated last month
- DeepSeek Native Sparse Attention pytorch implementation☆107Updated last month
- 模型压缩的小白入门教程☆22Updated last year
- Efficient, Flexible, and Highly Fault-Tolerant Model Service Management Based on SGLang☆59Updated last year
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆125Updated last week
- Models and examples built with OneFlow☆100Updated last year
- ☆149Updated 8 months ago
- Triton Documentation in Chinese Simplified / Triton 中文文档☆89Updated 6 months ago
- ☆56Updated last month
- ☆79Updated last year
- ☆69Updated 2 weeks ago
- simplify >2GB large onnx model☆64Updated 11 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆217Updated last year
- run ChatGLM2-6B in BM1684X☆50Updated last year
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆98Updated last year
- ☆79Updated 6 months ago