dhcode-cpp / online-softmaxLinks
simplest online-softmax notebook for explain Flash Attention
☆13Updated last year
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.☆31Updated 3 months ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆51Updated 2 years ago
- ☆51Updated 2 years ago
- Implementation of FlashAttention in PyTorch☆179Updated last year
- Transformer related optimization, including BERT, GPT☆17Updated 2 years ago
- qwen-nsa☆87Updated 2 months ago
- Inference code for LLaMA models☆128Updated 2 years ago
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆71Updated 2 years ago
- ☆79Updated 2 years ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆98Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆132Updated last year
- ☆150Updated 6 months ago
- Models and examples built with OneFlow☆101Updated last year
- Efficient, Flexible, and Highly Fault-Tolerant Model Service Management Based on SGLang☆61Updated last year
- ☆71Updated this week
- An easy-to-use package for implementing SmoothQuant for LLMs☆110Updated 9 months ago
- This is a personal reimplementation of Google's Infini-transformer, utilizing a small 2b model. The project includes both model and train…☆58Updated last year
- 模型压缩的小白入门教程☆22Updated last year
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆144Updated 2 weeks ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆49Updated 4 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆57Updated 2 years ago
- Efficient Mixture of Experts for LLM Paper List☆154Updated 3 months ago
- ☆153Updated 10 months ago
- from MHA, MQA, GQA to MLA by 苏剑林, with code☆36Updated 10 months ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated 2 years ago
- Triton Documentation in Chinese Simplified / Triton 中文文档☆96Updated 3 weeks ago
- ☆39Updated 10 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆23Updated last year
- Skywork-MoE: A Deep Dive into Training Techniques for Mixture-of-Experts Language Models☆138Updated last year
- Model compression toolkit engineered for enhanced usability, comprehensiveness, and efficiency.☆244Updated this week