daochenzha / autoshard
[KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems
☆21Updated 2 years ago
Alternatives and similar repositories for autoshard:
Users that are interested in autoshard 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
- Distributed Deep Graph Learning Framework for Dynamic Graphs☆12Updated last year
- Set of datasets for the deep learning recommendation model (DLRM).☆42Updated 2 years ago
- [NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems☆29Updated 2 years ago
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆61Updated 2 years ago
- Largest realworld open-source graph dataset - Worked done under IBM-Illinois Discovery Accelerator Institute and Amazon Research Awards a…☆79Updated 3 months ago
- ☆46Updated 2 years ago
- ☆19Updated 3 years ago
- Accelerating Recommender model training by leveraging popular choices -- VLDB 2022☆30Updated 6 months ago
- [ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhang…☆65Updated last year
- PyTorch-Direct code on top of PyTorch-1.8.0nightly (e152ca5) for Large Graph Convolutional Network Training with GPU-Oriented Data Commun…☆45Updated last year
- ☆15Updated 2 years ago
- [MLSys 2022] "BNS-GCN: Efficient Full-Graph Training of Graph Convolutional Networks with Partition-Parallelism and Random Boundary Node …☆54Updated last year
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆38Updated last year
- ICLR 2021☆47Updated 4 years ago
- Codes for SIGIR'22 Paper 'On-Device Next-Item Recommendation with Self-Supervised Knowledge Distillation'☆12Updated 2 years ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 3 years ago
- Source code for Learn Locally Correct Globally☆14Updated 3 years ago
- [ICLR 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Y…☆32Updated 2 years ago
- ☆13Updated last year
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆133Updated 2 years ago
- ☆45Updated this week
- ☆73Updated 3 years ago
- ☆16Updated 3 years ago
- [ NeurIPS '22 ] Data distillation for recommender systems. Shows equivalent performance with 2-3 orders less data.☆23Updated last year
- Enabling pure data parallel training of DLRM via caching and prefetching☆17Updated 3 years ago
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆38Updated 3 years ago
- [ICDCS 2023] Evaluation and Optimization of Gradient Compression for Distributed Deep Learning☆10Updated last year
- GraphZoom: A Multi-level Spectral Approach for Accurate and Scalable Graph Embedding☆114Updated 2 years ago
- ☆13Updated 4 years ago