dhcode-cpp / online-softmaxLinks
simplest online-softmax notebook for explain Flash Attention
☆11Updated 6 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:
- Implementation of FlashAttention in PyTorch☆155Updated 6 months ago
- ☢️ TensorRT 2023复赛——基于TensorRT-LLM的Llama模型推断加速优化☆49Updated last year
- ☆52Updated 2 years ago
- Inference code for LLaMA models☆122Updated last year
- ☆140Updated last week
- from MHA, MQA, GQA to MLA by 苏剑林, with code☆22Updated 4 months ago
- ☆36Updated last year
- A MoE impl for PyTorch, [ATC'23] SmartMoE☆64Updated 2 years ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆97Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆115Updated last year
- qwen-nsa☆68Updated 3 months ago
- TensorRT-in-Action 是一个 GitHub 代码库,提供了使用 TensorRT 的代码示例,并有对应 Jupyter Notebook。☆16Updated 2 years ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆102Updated 3 months ago
- ☆79Updated last year
- export llama to onnx☆129Updated 6 months ago
- an implementation of transformer, bert, gpt, and diffusion models for learning purposes☆155Updated 9 months ago
- ☆90Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆59Updated last year
- Transformer related optimization, including BERT, GPT☆17Updated last year
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆169Updated last year
- 模型压缩的小白入门教程☆22Updated last year
- The GPU RAM Estimator provides a simple tool for estimating GPU memory usage during training and inference.☆34Updated last year
- Models and examples built with OneFlow☆97Updated 9 months ago
- DeepSeek Native Sparse Attention pytorch implementation☆73Updated 4 months ago
- lightweighted deep learning inference service framework☆39Updated 4 years ago
- ☆47Updated 8 months ago
- ☆145Updated 4 months ago
- LLM101n: Let's build a Storyteller 中文版☆131Updated 11 months ago
- 从零到一实现了一个多模态大模型,并命名为Reyes(睿视),R:睿,eyes:眼。Reyes的参数量为8B,视觉编码器使用的是InternViT-300M-448px-V2_5,语言模型侧使用的是Qwen2.5-7B-Instruct,Reyes也通过一个两层MLP投影层连…☆22Updated 5 months ago
- 怎么训练一个LLM分词器☆151Updated 2 years ago