wolfecameron / GISTLinks
Repository for "GIST: Distributed training for large-scale graph convolutional networks"
☆15Updated 2 years ago
Alternatives and similar repositories for GIST
Users that are interested in GIST are comparing it to the libraries listed below
Sorting:
- [NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems☆29Updated 2 years ago
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago
- Implementation of Hyena Hierarchy in JAX☆10Updated 2 years ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆73Updated last year
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆36Updated last year
- Using FlexAttention to compute attention with different masking patterns☆47Updated last year
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)☆44Updated last year
- [NAACL 2025] Official Implementation of "HMT: Hierarchical Memory Transformer for Long Context Language Processing"☆75Updated 4 months ago
- Pytorch Implementation of Automatic Relation-aware Graph Network Proliferation (CVPR 2022 Oral)☆26Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆36Updated last year
- ☆10Updated last year
- Linear Attention Sequence Parallelism (LASP)☆87Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated last year
- Code repository for the public reproduction of the language modelling experiments on "MatFormer: Nested Transformer for Elastic Inference…☆28Updated last year
- ☆51Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated 2 years ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆100Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated last year
- Can GPT-4 Perform Neural Architecture Search?☆87Updated 2 years ago
- ☆26Updated last year
- Code for this paper "HyperRouter: Towards Efficient Training and Inference of Sparse Mixture of Experts via HyperNetwork"☆33Updated last year
- Advanced Ultra-Low Bitrate Compression Techniques for the LLaMA Family of LLMs☆110Updated last year
- Code repository for the paper - "AdANNS: A Framework for Adaptive Semantic Search"☆65Updated 2 years ago
- Defeating the Training-Inference Mismatch via FP16☆56Updated this week
- Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …☆114Updated last year
- KV Cache Steering for Inducing Reasoning in Small Language Models☆41Updated 3 months ago
- Low-Rank Llama Custom Training☆23Updated last year
- Utilities for Training Very Large Models☆58Updated last year
- Implementation of IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆25Updated 3 months ago
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆37Updated last year