IST-DASLab / MicroAdamLinks
This repository contains code for the MicroAdam paper.
☆19Updated 9 months ago
Alternatives and similar repositories for MicroAdam
Users that are interested in MicroAdam are comparing it to the libraries listed below
Sorting:
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆42Updated last year
- Work in progress.☆72Updated 2 months ago
- QuIP quantization☆59Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆93Updated 2 months ago
- An extention to the GaLore paper, to perform Natural Gradient Descent in low rank subspace☆17Updated 10 months ago
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)☆44Updated last year
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago
- ☆92Updated 3 weeks ago
- ☆126Updated 3 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆115Updated 2 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆87Updated last month
- ☆63Updated 5 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆129Updated 9 months ago
- ☆13Updated 8 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆185Updated 3 months ago
- [NeurIPS 2024] Low rank memory efficient optimizer without SVD☆31Updated 2 months ago
- Beyond KV Caching: Shared Attention for Efficient LLMs☆19Updated last year
- Code for "RSQ: Learning from Important Tokens Leads to Better Quantized LLMs"☆19Updated 3 months ago
- Fast and memory-efficient exact attention☆69Updated 6 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated 11 months ago
- From GaLore to WeLore: How Low-Rank Weights Non-uniformly Emerge from Low-Rank Gradients. Ajay Jaiswal, Lu Yin, Zhenyu Zhang, Shiwei Liu,…☆48Updated 4 months ago
- PB-LLM: Partially Binarized Large Language Models☆153Updated last year
- ☆23Updated 5 months ago
- Here we will test various linear attention designs.☆62Updated last year
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆40Updated last year
- Kinetics: Rethinking Test-Time Scaling Laws☆80Updated 2 months ago
- FlexAttention w/ FlashAttention3 Support☆27Updated 11 months ago
- ☆69Updated last year
- ☆29Updated 10 months ago
- Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training☆27Updated 2 months ago