wolfecameron / GIST
Repository for "GIST: Distributed training for large-scale graph convolutional networks"
☆14Updated last year
Related projects: ⓘ
- The implementation for MLSys 2023 paper: "Cuttlefish: Low-rank Model Training without All The Tuning"☆42Updated last year
- [NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems☆28Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆77Updated last year
- ☆13Updated this week
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆22Updated 3 months ago
- This repository contains code for the MicroAdam paper.☆9Updated 2 months ago
- Implementation of Hyena Hierarchy in JAX☆10Updated last year
- Using FlexAttention to compute attention with different masking patterns☆28Updated last week
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆28Updated 4 months ago
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆37Updated last year
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆29Updated 7 months ago
- Low-Rank Llama Custom Training☆16Updated 5 months ago
- ☆16Updated 2 years ago
- Lottery Ticket Adaptation☆33Updated 3 weeks ago
- The Efficiency Spectrum of LLM☆50Updated 9 months ago
- ☆29Updated 3 weeks ago
- ICLR 2021☆42Updated 3 years ago
- Code for "Merging Text Transformers from Different Initializations"☆18Updated last month
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆29Updated 6 months ago
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆31Updated 2 years ago
- Code repository for the public reproduction of the language modelling experiments on "MatFormer: Nested Transformer for Elastic Inference…☆17Updated 10 months ago
- Experiments for "A Closer Look at In-Context Learning under Distribution Shifts"☆20Updated last year
- ☆12Updated 9 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆43Updated 3 weeks ago
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆68Updated 3 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆36Updated 8 months ago
- GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU☆17Updated 3 weeks ago
- Codes of the paper Deformable Butterfly: A Highly Structured and Sparse Linear Transform.☆12Updated 2 years ago
- [ACL 2024] Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Models☆59Updated 3 months ago