The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS 2025]
☆60Oct 11, 2024Updated last year
Alternatives and similar repositories for prepacking
Users that are interested in prepacking are comparing it to the libraries listed below
Sorting:
- ☆15Nov 5, 2024Updated last year
- ☆13Jun 22, 2025Updated 8 months ago
- Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…☆157Apr 7, 2025Updated 11 months ago
- ☆21Sep 6, 2021Updated 4 years ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆180Jul 12, 2024Updated last year
- Source code of the paper: RetrievalQA: Assessing Adaptive Retrieval-Augmented Generation for Short-form Open-Domain Question Answering [F…☆68May 28, 2024Updated last year
- ☆13Jul 2, 2025Updated 8 months ago
- Bunch of notebooks for pre-training custom Saiga-like LLM☆12Feb 9, 2024Updated 2 years ago
- ☆29Jan 23, 2024Updated 2 years ago
- This project is concerned with my participating in the RuNNE competition https://github.com/dialogue-evaluation/RuNNE☆13Jun 28, 2023Updated 2 years ago
- ☆15Aug 18, 2022Updated 3 years ago
- The official github repo for the open online courses: "Dive into LLMs".☆10Mar 15, 2024Updated 2 years ago
- vLLM performance dashboard☆43Apr 26, 2024Updated last year
- (WIP) Parallel inference for black-forest-labs' FLUX model.☆19Nov 18, 2024Updated last year
- Implementation of Decision Stacks: Flexible RL via Modular Generative Models [NeurIPS 2023]☆12Jun 27, 2023Updated 2 years ago
- MedMax: Mixed-Modal Instruction Tuning for Training Biomedical Assistants☆42Sep 19, 2025Updated 6 months ago
- My fork os allen AI's OLMo for educational purposes.☆28Dec 5, 2024Updated last year
- [NeurIPS 2024] Goldfish Loss: Mitigating Memorization in Generative LLMs☆97Nov 17, 2024Updated last year
- Lightweight Python Wrapper for OpenVINO, enabling LLM inference on NPUs☆27Dec 17, 2024Updated last year
- 🚢 Data Toolkit for Sailor Language Models☆96Feb 24, 2025Updated last year
- Original code base for On Pretraining Data Diversity for Self-Supervised Learning☆14Dec 30, 2024Updated last year
- ☆14Nov 20, 2022Updated 3 years ago
- Pytorch/XLA SPMD Test code in Google TPU☆23Apr 3, 2024Updated last year
- Repo for "Zemi: Learning Zero-Shot Semi-Parametric Language Models from Multiple Tasks" ACL 2023 Findings☆15May 3, 2023Updated 2 years ago
- A model serving framework for various research and production scenarios. Seamlessly built upon the PyTorch and HuggingFace ecosystem.☆23Oct 11, 2024Updated last year
- ☆15Jun 2, 2025Updated 9 months ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆48Jun 19, 2024Updated last year
- This is the official repo for the paper "Accelerating Parallel Sampling of Diffusion Models" Tang et al. ICML 2024 https://openreview.net…☆16Jul 19, 2024Updated last year
- Work in progress.☆79Nov 25, 2025Updated 3 months ago
- Quantized Attention on GPU☆44Nov 22, 2024Updated last year
- ☆27Jul 20, 2024Updated last year
- WIP: project for engineering automatic bot (chatbot mainly)☆13Sep 3, 2023Updated 2 years ago
- a simple API to use CUPTI☆10Aug 19, 2025Updated 7 months ago
- ☆105Sep 9, 2024Updated last year
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 2 years ago
- official code for paper Probing the Decision Boundaries of In-context Learning in Large Language Models. https://arxiv.org/abs/2406.11233…☆19Jul 27, 2025Updated 7 months ago
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated last year
- annotated-transformer-kr☆15May 16, 2019Updated 6 years ago
- "Learning-based One-line intelligence Owner Network Connectivity Tool"☆15Apr 19, 2023Updated 2 years ago