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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 3 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
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 4 years ago
- Large-scale model inference.☆629Sep 12, 2023Updated 3 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆18Nov 1, 2021Updated 4 years ago
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- Documentation for Colossal-AI☆24Jun 6, 2025Updated last year
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 3 years ago
- Scalable GPU Kernel Fission/Fusion Transformation for Memory-Bound Kernels☆14Aug 26, 2015Updated 11 years ago
- ☆12Apr 30, 2024Updated 2 years ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- A Web-Based Visualization Tool for Biclustering of Multivariate Time Series☆10Feb 17, 2023Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆46Jul 19, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆208May 22, 2026Updated 4 months ago
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,073Aug 3, 2026Updated 2 months ago
- An implementation of online data mixing for the Pile dataset, based on the GPT-NeoX library.☆14Jan 9, 2024Updated 2 years ago
- ☆22Nov 20, 2020Updated 5 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆134Sep 15, 2022Updated 4 years ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆411Updated this week
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- A simple demonstration of how PyTorch autograd works☆16Sep 23, 2021Updated 5 years ago
- Manages vllm-nccl dependency☆19Jun 3, 2024Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Oct 27, 2022Updated 3 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 years ago
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆17Aug 4, 2022Updated 4 years ago
- Pytorch domain library for recommendation systems☆2,610Updated this week
- Code for paper "Engineering a High-Performance GPU B-Tree" accepted to PPoPP 2019☆57Jun 27, 2022Updated 4 years ago
- End to End steps for adding custom ops in PyTorch.☆24Aug 20, 2020Updated 6 years ago
- Visualize the Expert Parallelism Load Balancer☆17Mar 15, 2025Updated last year
- Code repository for the SOSP'25 paper DCP: Addressing Input Dynamism In Long-Context Training via Dynamic Context Parallelism.☆23Nov 28, 2025Updated 10 months ago
- A simple C++11 read-write lock.☆10Aug 19, 2018Updated 8 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆30Sep 4, 2023Updated 3 years ago
- A parameter server implement with MPI.☆11Nov 15, 2017Updated 8 years ago
- SoCC'20 and TPDS'21: Scaling GNN Training on Large Graphs via Computation-aware Caching and Partitioning.☆51May 23, 2023Updated 3 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆133Jul 6, 2023Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆166Aug 19, 2026Updated last month
- High performance RDMA-based distributed feature collection component for training GNN model on EXTREMELY large graph☆55Jul 3, 2022Updated 4 years ago
- ☆16Sep 4, 2023Updated 3 years ago