daochenzha / dreamshard
[NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems
☆29Updated last year
Alternatives and similar repositories for dreamshard:
Users that are interested in dreamshard are comparing it to the libraries listed below
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated last year
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆21Updated last year
- Can GPT-4 Perform Neural Architecture Search?☆85Updated last year
- ☆19Updated 2 years ago
- [ NeurIPS '22 ] Data distillation for recommender systems. Shows equivalent performance with 2-3 orders less data.☆23Updated last year
- pytorch open-source library for the paper "AdaTT Adaptive Task-to-Task Fusion Network for Multitask Learning in Recommendations"☆44Updated 6 months ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆42Updated 2 years ago
- The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"☆15Updated 11 months ago
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆40Updated 8 months ago
- AutoLossGen: Automatic Loss Function Generation for Recommender Systems☆22Updated 2 years ago
- Official codebase for NeurIPS 2022 paper End-to-end Learning to Index and Search in Large Output Spaces☆12Updated last year
- ☆18Updated 2 years ago
- Repository for "GIST: Distributed training for large-scale graph convolutional networks"☆14Updated 2 years ago
- RecZilla: Metalearning for algorithm selection on Recommender Systems☆23Updated last year
- ☆25Updated last year
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆38Updated 3 years ago
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆32Updated last year
- Code for the Population-Based Bandits Algorithm, presented at NeurIPS 2020.☆20Updated 3 years ago
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 5 months ago
- Retrieval with Learned Similarities (http://arxiv.org/abs/2407.15462, WWW'25 Oral)☆37Updated 3 weeks ago
- Hyperparameter tuning via uncertainty modeling☆47Updated 10 months ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated last month
- ☆33Updated 7 months ago
- [KDD 2021, Research Track] DiffMG: Differentiable Meta Graph Search for Heterogeneous Graph Neural Networks☆30Updated 3 years ago
- Towards LLM Empowered Recommendation via Tool Learning☆15Updated 9 months ago
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆32Updated 10 months ago
- Code for EMNLP'24 paper - On Diversified Preferences of Large Language Model Alignment☆15Updated 7 months ago
- ☆11Updated 11 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Updated last year
- ☆28Updated 4 months ago