daochenzha / dreamshardLinks
[NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems
☆29Updated 2 years ago
Alternatives and similar repositories for dreamshard
Users that are interested in dreamshard are comparing it to the libraries listed below
Sorting:
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆36Updated last year
- Can GPT-4 Perform Neural Architecture Search?☆88Updated 2 years ago
- Code for COMET: Cardinality Constrained Mixture of Experts with Trees and Local Search☆11Updated 2 years ago
- Hyperparameter tuning via uncertainty modeling☆49Updated last year
- ☆10Updated 2 years ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆48Updated 3 years ago
- Repository for "GIST: Distributed training for large-scale graph convolutional networks"☆15Updated 2 years ago
- Using FlexAttention to compute attention with different masking patterns☆47Updated last year
- Official codebase for NeurIPS 2022 paper End-to-end Learning to Index and Search in Large Output Spaces☆12Updated 2 years ago
- Linear Attention Sequence Parallelism (LASP)☆88Updated last year
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated last year
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆38Updated last year
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆46Updated 2 years ago
- Lottery Ticket Adaptation☆40Updated last year
- Implementation of a Hierarchical Mamba as described in the paper: "Hierarchical State Space Models for Continuous Sequence-to-Sequence Mo…☆14Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Updated 2 years ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆69Updated last year
- [ACL 2025 Main] Repository for the paper: 500xCompressor: Generalized Prompt Compression for Large Language Models☆54Updated 7 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆85Updated 6 months ago
- A testbed for agents and environments that can automatically improve models through data generation.☆27Updated 10 months ago
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Updated 3 years ago
- Embroid: Unsupervised Prediction Smoothing Can Improve Few-Shot Classification☆11Updated 2 years ago
- ☆52Updated last year
- ☆34Updated 3 months ago
- ☆57Updated last year
- ☆35Updated last year
- A Learnable LSH Framework for Efficient NN Training☆34Updated 4 years ago