kernelmachine / balanced-kmeansLinks
☆15Updated last year
Alternatives and similar repositories for balanced-kmeans
Users that are interested in balanced-kmeans are comparing it to the libraries listed below
Sorting:
- [ICML 2024] Junk DNA Hypothesis: A Task-Centric Angle of LLM Pre-trained Weights through Sparsity; Lu Yin*, Ajay Jaiswal*, Shiwei Liu, So…☆16Updated 2 months ago
- Using FlexAttention to compute attention with different masking patterns☆44Updated 9 months ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated last year
- Long Context Extension and Generalization in LLMs☆57Updated 9 months ago
- ☆74Updated last year
- An evaluation suite for Retrieval-Augmented Generation (RAG).☆20Updated last month
- This package implements THOR: Transformer with Stochastic Experts.☆65Updated 3 years ago
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆52Updated 2 years ago
- ☆32Updated last year
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆60Updated 8 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆66Updated 6 months ago
- A Kernel-Based View of Language Model Fine-Tuning https://arxiv.org/abs/2210.05643☆75Updated last year
- Beyond KV Caching: Shared Attention for Efficient LLMs☆19Updated 11 months ago
- Codebase for Context-aware Meta-learned Loss Scaling (CaMeLS). https://arxiv.org/abs/2305.15076.☆25Updated last year
- This repo is based on https://github.com/jiaweizzhao/GaLore☆28Updated 9 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- [ICML2024 Spotlight] Fine-Tuning Pre-trained Large Language Models Sparsely☆23Updated 11 months ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆47Updated 2 years ago
- ☆96Updated 8 months ago
- ☆47Updated last week
- ☆39Updated 2 years ago
- ☆137Updated 11 months ago
- [ICML‘2024] "LoCoCo: Dropping In Convolutions for Long Context Compression", Ruisi Cai, Yuandong Tian, Zhangyang Wang, Beidi Chen☆16Updated 9 months ago
- ☆29Updated 11 months ago
- Test-time-training on nearest neighbors for large language models☆41Updated last year
- Pytorch implementation for "Compressed Context Memory For Online Language Model Interaction" (ICLR'24)☆61Updated last year
- ☆57Updated last year
- Repo for ACL2023 Findings paper "Emergent Modularity in Pre-trained Transformers"☆25Updated 2 years ago
- Running inference on the ZeroSCROLLS benchmark☆20Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆76Updated last year