A memory efficient DLRM training solution using ColossalAI
☆107Nov 22, 2022Updated 3 years ago
Alternatives and similar repositories for CachedEmbedding
Users that are interested in CachedEmbedding are comparing it to the libraries listed below
Sorting:
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 2 years ago
- PSTensor provides a way to hack the memory management of tensors in TensorFlow and PyTorch by defining your own C++ Tensor Class.☆10Feb 10, 2022Updated 4 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Nov 27, 2024Updated last year
- Artifact of ASPLOS'23 paper entitled: GRACE: A Scalable Graph-Based Approach to Accelerating Recommendation Model Inference☆19Mar 5, 2023Updated 3 years ago
- GPT Demo with hybrid distributed training☆10Dec 1, 2022Updated 3 years ago
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- Large-scale model inference.☆627Sep 12, 2023Updated 2 years ago
- ☆31Feb 22, 2024Updated 2 years ago
- ☆18Nov 1, 2021Updated 4 years ago
- Poise source code repo☆12Aug 12, 2020Updated 5 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 2 years ago
- ☆12Apr 30, 2024Updated last year
- Torch Distributed Experimental☆117Aug 5, 2024Updated last year
- Scalable GPU Kernel Fission/Fusion Transformation for Memory-Bound Kernels☆14Aug 26, 2015Updated 10 years ago
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,052Mar 12, 2026Updated last week
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 8 months ago
- A collection of models built with ColossalAI☆32Nov 22, 2022Updated 3 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆197Feb 27, 2026Updated 3 weeks ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆411Jun 14, 2025Updated 9 months ago
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated 7 months ago
- Sharing the codebase and steps for artifact evaluation for ISCA 2023 paper☆15Feb 20, 2024Updated 2 years ago
- ☆22Nov 20, 2020Updated 5 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆135Sep 15, 2022Updated 3 years ago
- Accelerating Recommender model training by leveraging popular choices -- VLDB 2022☆31Sep 15, 2024Updated last year
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- [ICLR 2022] "PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication" by Cheng Wan, Y…☆34Mar 15, 2023Updated 3 years ago
- Set of datasets for the deep learning recommendation model (DLRM).☆49Dec 21, 2022Updated 3 years ago
- Pytorch domain library for recommendation systems☆2,490Updated this week
- Manages vllm-nccl dependency☆17Jun 3, 2024Updated last year
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Oct 27, 2022Updated 3 years ago
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆17Aug 4, 2022Updated 3 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 years ago
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆302Aug 17, 2023Updated 2 years ago
- Code for "Baleen: ML Admission & Prefetching for Flash Caches" (FAST 2024).☆27Feb 29, 2024Updated 2 years ago
- End to End steps for adding custom ops in PyTorch.☆24Aug 20, 2020Updated 5 years ago
- ☆30Sep 4, 2023Updated 2 years ago
- SoCC'20 and TPDS'21: Scaling GNN Training on Large Graphs via Computation-aware Caching and Partitioning.☆51May 23, 2023Updated 2 years ago
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆55Jul 3, 2022Updated 3 years ago