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)☆35Updated last year
- Can GPT-4 Perform Neural Architecture Search?☆87Updated 2 years ago
- Repository for "GIST: Distributed training for large-scale graph convolutional networks"☆15Updated 2 years ago
- The code implementation of MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models…☆36Updated last year
- Learning from preferences is a common paradigm for fine-tuning language models. Yet, many algorithmic design decisions come into play. Ou…☆30Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated last year
- Using FlexAttention to compute attention with different masking patterns☆44Updated 10 months ago
- AutoMoE: Neural Architecture Search for Efficient Sparsely Activated Transformers☆47Updated 2 years ago
- Lottery Ticket Adaptation☆39Updated 8 months ago
- This is the repository holding code and data for "FrugalML: How to Use ML Prediction APIs More Accurately and Cheaply".☆35Updated 4 years ago
- ☆10Updated last year
- Code for COMET: Cardinality Constrained Mixture of Experts with Trees and Local Search☆11Updated 2 years ago
- [KDD 2022] AutoShard: Automated Embedding Table Sharding 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 2 years ago
- ☆35Updated 8 months ago
- Linear Attention Sequence Parallelism (LASP)☆85Updated last year
- Repository for NPHardEval, a quantified-dynamic benchmark of LLMs☆57Updated last year
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 11 months ago
- Hyperparameter tuning via uncertainty modeling☆47Updated last year
- Understanding the correlation between different LLM benchmarks☆29Updated last year
- A Data Source for Reasoning Embodied Agents☆19Updated last year
- Repository for CPU Kernel Generation for LLM Inference☆26Updated 2 years ago
- Dateset Reset Policy Optimization☆30Updated last year
- Official Code Repository for EnvGen: Generating and Adapting Environments via LLMs for Training Embodied Agents (COLM 2024)☆35Updated last year
- Official repository for the paper "Zero-Shot AutoML with Pretrained Models"☆47Updated last year
- ☆29Updated 2 years ago
- Implementation of Hyena Hierarchy in JAX☆10Updated 2 years ago
- A testbed for agents and environments that can automatically improve models through data generation.☆27Updated 5 months ago
- The repository contains code for Adaptive Data Optimization☆25Updated 8 months ago
- ☆18Updated last week