daochenzha / autoshardLinks
[KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems
☆22Updated 2 years ago
Alternatives and similar repositories for autoshard
Users that are interested in autoshard are comparing it to the libraries listed below
Sorting:
- Set of datasets for the deep learning recommendation model (DLRM).☆48Updated 2 years ago
- Distributed Deep Graph Learning Framework for Dynamic Graphs☆18Updated last year
- Largest realworld open-source graph dataset - Worked done under IBM-Illinois Discovery Accelerator Institute and Amazon Research Awards a…☆85Updated 5 months ago
- ICLR 2021☆48Updated 4 years ago
- ☆21Updated 3 years ago
- [MLSys 2022] "BNS-GCN: Efficient Full-Graph Training of Graph Convolutional Networks with Partition-Parallelism and Random Boundary Node …☆56Updated 2 years ago
- Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB)☆44Updated 2 years ago
- A GPU-accelerated graph learning library for PyTorch, facilitating the scaling of GNN training and inference.☆146Updated 2 months ago
- [NeurIPS 2022] DreamShard: Generalizable Embedding Table Placement for Recommender Systems☆29Updated 2 years ago
- [ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhang…☆68Updated 2 years ago
- Codes for SIGIR'22 Paper 'On-Device Next-Item Recommendation with Self-Supervised Knowledge Distillation'☆14Updated 2 years ago
- ☆14Updated 4 years ago
- [ICLR 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Y…☆33Updated 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…☆59Updated 3 years ago
- ☆216Updated last year
- Accelerating Recommender model training by leveraging popular choices -- VLDB 2022☆31Updated last year
- Official implementation of our VQ-GNN paper (NeurIPS2021)☆38Updated 4 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆302Updated 2 years ago
- ☆47Updated 3 years ago
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16Updated 2 years ago
- Enabling pure data parallel training of DLRM via caching and prefetching☆17Updated 4 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆135Updated 3 years ago
- [NeurIPS 2021]: Improve the GNN expressivity and scalability by decoupling the depth and receptive field of state-of-the-art GNN architec…☆137Updated 3 years ago
- ☆11Updated 2 years ago
- ☆70Updated 4 years ago
- Graph Partitoning Using Graph Convolutional Networks☆66Updated 3 years ago
- A scalable graph learning toolkit for extremely large graph datasets. (WWW'22, 🏆 Best Student Paper Award)☆157Updated last year
- The official SALIENT system described in the paper "Accelerating Training and Inference of Graph Neural Networks with Fast Sampling and P…☆40Updated 2 years ago
- Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch☆165Updated 3 years ago
- ☆49Updated 8 months ago