daochenzha / autoshard
[KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems
☆21Updated last year
Related projects ⓘ
Alternatives and complementary repositories for autoshard
- Set of datasets for the deep learning recommendation model (DLRM).☆41Updated last year
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated last year
- [NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems☆28Updated last year
- ☆18Updated 2 years ago
- Accelerating Recommender model training by leveraging popular choices -- VLDB 2022☆29Updated 2 months ago
- Largest realworld open-source graph dataset - Worked done under IBM-Illinois Discovery Accelerator Institute and Amazon Research Awards a…☆76Updated 2 months ago
- Distributed Deep Graph Learning Framework for Dynamic Graphs☆11Updated 7 months ago
- ICLR 2021☆44Updated 3 years ago
- [ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhang…☆63Updated last year
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆33Updated 3 years ago
- 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 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Y…☆31Updated last year
- ☆45Updated 2 years ago
- [CIKM 2022] Towards Automated Over-Sampling for Imbalanced Classification☆10Updated last year
- [MLSys 2022] "BNS-GCN: Efficient Full-Graph Training of Graph Convolutional Networks with Partition-Parallelism and Random Boundary Node …☆52Updated last year
- Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into ef…☆60Updated 2 years ago
- The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"☆15Updated 8 months ago
- ☆12Updated last year
- ☆72Updated 3 years ago
- Source code for Learn Locally Correct Globally☆14Updated 3 years ago
- Enabling pure data parallel training of DLRM via caching and prefetching☆18Updated 3 years ago
- 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
- ☆34Updated 5 months ago
- The released code for the paper: Search to aggregate neighborhood for graph neural network, in ICDE 2021.☆21Updated 2 years ago
- ☆52Updated 9 months ago
- ☆15Updated 2 years ago
- Retrieval with Learned Similarities☆15Updated this week
- Pytorch implementation of differentiable group normalization (NeurIPS 2020)☆37Updated 4 years ago
- Surrogate-based Hyperparameter Tuning System☆27Updated last year
- ☆12Updated 3 years ago