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
- A collection of models built with ColossalAI☆32Nov 22, 2022Updated 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
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- [SIGMOD 2024] CAFE: Towards Compact, Adaptive, and Fast Embedding for Large-scale Recommendation Models☆13Mar 19, 2025Updated last year
- Large-scale model inference.☆629Sep 12, 2023Updated 3 years ago
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- Documentation for Colossal-AI☆24Jun 6, 2025Updated last year
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Mar 24, 2023Updated 3 years ago
- Poise source code repo☆12Aug 12, 2020Updated 6 years ago
- ☆18Nov 1, 2021Updated 4 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 3 years ago
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Sharing the codebase and steps for artifact evaluation for ISCA 2023 paper☆16Feb 20, 2024Updated 2 years ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆410Updated this week
- Scalable GPU Kernel Fission/Fusion Transformation for Memory-Bound Kernels☆14Aug 26, 2015Updated 11 years ago
- ☆12Apr 30, 2024Updated 2 years ago
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,071Aug 3, 2026Updated 2 months ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆208Updated this week
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆134Sep 15, 2022Updated 4 years ago
- Pytorch domain library for recommendation systems☆2,619Updated this week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Containerization and deployment scripts for remote-index-server and workflows to generate monolithic index for github.com/llvm/llvm-proje…☆25Updated this week
- PyTorch Library for Low-Latency, High-Throughput Graph Learning on GPUs.☆305Aug 17, 2023Updated 3 years ago
- Compute free-support Wasserstein barycenters exactly☆10Aug 22, 2024Updated 2 years ago
- An implementation of online data mixing for the Pile dataset, based on the GPT-NeoX library.☆15Jan 9, 2024Updated 2 years ago
- A benchmarking suite for heterogeneous systems. The primary goal of this project is to improve and update aspects of existing benchmarkin…☆44Jan 30, 2026Updated 9 months ago
- ☆22Nov 20, 2020Updated 5 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- An external memory allocator example for PyTorch.☆16Aug 10, 2025Updated last year
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,597Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Oct 27, 2022Updated 4 years ago
- Tracking Books of TUNA Library☆12Nov 17, 2020Updated 5 years ago
- Official implementation of Graph Neural Networks as Gradient Flows.☆15Aug 15, 2023Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆165Aug 19, 2026Updated 2 months ago
- Dorylus: Affordable, Scalable, and Accurate GNN Training☆77May 31, 2021Updated 5 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 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