daochenzha / dreamshard
[NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems
☆28Updated last year
Related projects ⓘ
Alternatives and complementary repositories for dreamshard
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆21Updated last year
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆79Updated last year
- Repository for "GIST: Distributed training for large-scale graph convolutional networks"☆14Updated last year
- ☆18Updated 2 years ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆42Updated 2 years ago
- [ NeurIPS '22 ] Data distillation for recommender systems. Shows equivalent performance with 2-3 orders less data.☆22Updated last year
- ☆30Updated 2 weeks ago
- ☆12Updated 2 years ago
- ICLR 2021☆44Updated 3 years ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆29Updated 6 months ago
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆33Updated 3 years ago
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆37Updated 5 months ago
- pytorch open-source library for the paper "AdaTT Adaptive Task-to-Task Fusion Network for Multitask Learning in Recommendations"☆42Updated 3 months ago
- [EMNLP 2024 Main] Virtual Personas for Language Models via an Anthology of Backstories☆18Updated this week
- [CIKM 2022] Towards Automated Over-Sampling for Imbalanced Classification☆10Updated last year
- Lottery Ticket Adaptation☆36Updated last month
- Official codebase for NeurIPS 2022 paper End-to-end Learning to Index and Search in Large Output Spaces☆12Updated last year
- ☆26Updated last year
- ☆24Updated 7 months ago
- ☆21Updated last year
- Towards LLM Empowered Recommendation via Tool Learning☆14Updated 5 months ago
- ☆28Updated 2 years ago
- Can GPT-4 Perform Neural Architecture Search?☆82Updated last year
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 2 months ago
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆41Updated 10 months ago
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆30Updated 9 months ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks"☆14Updated 3 weeks ago
- Repository for NPHardEval, a quantified-dynamic benchmark of LLMs☆48Updated 7 months ago