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:
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated 2 years ago
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Updated 2 years ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆34Updated last year
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆19Updated last month
- Can GPT-4 Perform Neural Architecture Search?☆87Updated 2 years ago
- Hyperparameter tuning via uncertainty modeling☆47Updated last year
- ☆14Updated 7 months ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆47Updated 2 years ago
- Using FlexAttention to compute attention with different masking patterns☆44Updated 9 months ago
- ☆32Updated 8 months ago
- ☆22Updated 9 months ago
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated last year
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆35Updated last year
- AutoLossGen: Automatic Loss Function Generation for Recommender Systems☆22Updated 3 years ago
- ☆51Updated last year
- ☆23Updated 2 years ago
- Linear Attention Sequence Parallelism (LASP)☆85Updated last year
- Repository for "GIST: Distributed training for large-scale graph convolutional networks"☆15Updated 2 years ago
- Kinetics: Rethinking Test-Time Scaling Laws☆65Updated this week
- Efficient Scaling laws and collaborative pretraining.☆16Updated 5 months ago
- Lottery Ticket Adaptation☆39Updated 7 months ago
- ☆45Updated last year
- ☆18Updated last month
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated last year
- A testbed for agents and environments that can automatically improve models through data generation.☆24Updated 4 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆66Updated 7 months ago
- Graph Transformers for Large Graphs☆21Updated last year
- MACTA: A Multi-agent Reinforcement Learning Approach for Cache Timing Attacks and Detection☆46Updated 2 years ago
- The repository provides code for the paper RECE: Reduced Cross-Entropy Loss for Large-Catalogue Sequential Recommenders, CIKM'24☆11Updated 8 months ago
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago