thu-ml / low-bit-optimizers
Low-bit optimizers for PyTorch
☆125Updated last year
Alternatives and similar repositories for low-bit-optimizers:
Users that are interested in low-bit-optimizers are comparing it to the libraries listed below
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆120Updated last month
- ☆115Updated 2 weeks ago
- 🔥 A minimal training framework for scaling FLA models☆70Updated this week
- ☆143Updated last year
- Official implementation of "DoRA: Weight-Decomposed Low-Rank Adaptation"☆124Updated 10 months ago
- ☆217Updated 8 months ago
- An algorithm for static activation quantization of LLMs☆119Updated last month
- ☆41Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆99Updated 8 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆206Updated 6 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆201Updated 3 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆57Updated 4 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆157Updated 7 months ago
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆35Updated 8 months ago
- Unofficial implementations of block/layer-wise pruning methods for LLMs.☆63Updated 10 months ago
- ☆224Updated 10 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆27Updated 11 months ago
- ☆125Updated last year
- QuIP quantization☆51Updated 11 months ago
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆148Updated 8 months ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆94Updated last week
- ☆75Updated last month
- Implementation of Speculative Sampling as described in "Accelerating Large Language Model Decoding with Speculative Sampling" by Deepmind☆90Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆65Updated 5 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆85Updated 4 months ago
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆49Updated 8 months ago
- The official implementation of the EMNLP 2023 paper LLM-FP4☆185Updated last year
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆91Updated last year
- Reorder-based post-training quantization for large language model☆185Updated last year