High performance distributed framework for training deep learning recommendation models based on PyTorch.
☆411Jun 14, 2025Updated 8 months ago
Alternatives and similar repositories for PERSIA
Users that are interested in PERSIA are comparing it to the libraries listed below
Sorting:
- Bagua Speeds up PyTorch☆884Aug 1, 2024Updated last year
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,048Sep 15, 2025Updated 5 months ago
- A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster☆161Apr 20, 2024Updated last year
- Pytorch domain library for recommendation systems☆2,477Updated this week
- DeepRec is a high-performance recommendation deep learning framework based on TensorFlow. It is hosted in incubation in LF AI & Data Foun…☆1,168Jan 21, 2025Updated last year
- 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 high performance and generic framework for distributed DNN training☆3,716Oct 3, 2023Updated 2 years ago
- High performance NCCL plugin for Bagua.☆15Sep 15, 2021Updated 4 years ago
- Large batch training of CTR models based on DeepCTR with CowClip.☆173Feb 8, 2023Updated 3 years ago
- Additional utils and helpers to extend TensorFlow when build recommendation systems, contributed and maintained by SIG Recommenders.☆632Sep 4, 2025Updated 6 months ago
- Core communication lib for Bagua.☆48Sep 15, 2021Updated 4 years ago
- a TensorFlow-based distributed training framework optimized for large-scale sparse data.☆333Dec 23, 2025Updated 2 months ago
- deepx_core是一个专注于张量计算/深度学习的基础库☆380Apr 15, 2025Updated 10 months ago
- Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training☆1,863Updated this week
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆194Updated this week
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- ☆251Jul 25, 2024Updated last year
- The DGL Operator makes it easy to run Deep Graph Library (DGL) graph neural network training on Kubernetes☆44Sep 15, 2021Updated 4 years ago
- An implementation of a deep learning recommendation model (DLRM)☆4,020Jan 12, 2026Updated last month
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆20Feb 23, 2024Updated 2 years ago
- TensorFlow Recommenders is a library for building recommender system models using TensorFlow.☆2,000Jan 23, 2026Updated last month
- An industrial deep learning framework for high-dimension sparse data☆4,306Sep 25, 2024Updated last year
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4☆969Dec 21, 2025Updated 2 months ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- ☆219Aug 17, 2023Updated 2 years ago
- Deep Matching, Correlation and Prediction (DeepMCP) Model☆124Jul 22, 2020Updated 5 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,535Updated this week
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆193Jul 20, 2022Updated 3 years ago
- Resource-adaptive cluster scheduler for deep learning training.☆454Mar 5, 2023Updated 3 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆284Dec 17, 2025Updated 2 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,184Dec 9, 2023Updated 2 years ago
- a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.☆1,544Jul 18, 2025Updated 7 months ago
- Mesh TensorFlow: Model Parallelism Made Easier☆1,625Nov 17, 2023Updated 2 years ago
- A lightweight parameter server interface☆1,560Jan 11, 2023Updated 3 years ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Bagua tutorials.☆13Sep 4, 2022Updated 3 years ago
- A fast MoE impl for PyTorch☆1,840Feb 10, 2025Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆164Jan 12, 2026Updated last month
- The road to hack SysML and become an system expert☆510Sep 25, 2024Updated last year