dhcode-cpp / online-softmaxLinks
simplest online-softmax notebook for explain Flash Attention
☆13Updated 8 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 this week
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆50Updated last year
- ☆53Updated 2 years ago
- Inference code for LLaMA models☆123Updated 2 years ago
- Implementation of FlashAttention in PyTorch☆168Updated 8 months ago
- from MHA, MQA, GQA to MLA by 苏剑林, with code☆29Updated 7 months ago
- qwen-nsa☆74Updated 5 months ago
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆122Updated last year
- ☆143Updated 2 months ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆15Updated 2 years ago
- lightweighted deep learning inference service framework☆40Updated 4 years ago
- 模型压缩的小白入门教程☆22Updated last year
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆70Updated 2 years ago
- Triton Documentation in Chinese Simplified / Triton 中文文档☆82Updated 5 months ago
- simplify >2GB large onnx model☆63Updated 9 months ago
- 青稞Talk☆145Updated last week
- ☆79Updated last year
- Transformer related optimization, including BERT, GPT☆17Updated 2 years ago
- Efficient Mixture of Experts for LLM Paper List☆127Updated this week
- ☆33Updated 6 months ago
- Efficient, Flexible, and Highly Fault-Tolerant Model Service Management Based on SGLang☆56Updated 10 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆111Updated 5 months ago
- RLHF experiments on a single A100 40G GPU. Support PPO, GRPO, REINFORCE, RAFT, RLOO, ReMax, DeepSeek R1-Zero reproducing.☆72Updated 7 months ago
- 怎么训练一个LLM分词器☆152Updated 2 years ago
- ☆60Updated last week
- ☆147Updated 6 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆106Updated 5 months ago
- Tutorial for Ray☆29Updated last year
- run ChatGLM2-6B in BM1684X☆49Updated last year
- ☆37Updated 2 years ago