This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as recommendation and natural language processing. We showed this library can reduce the total model size by up to 100x in Facebook’s open sourced DLRM model while achieving same model quality. Our implementation …
☆194Jul 20, 2022Updated 3 years ago
Alternatives and similar repositories for FBTT-Embedding
Users that are interested in FBTT-Embedding are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…☆29Oct 12, 2021Updated 4 years ago
- An implementation of a deep learning recommendation model (DLRM)☆4,027Jan 12, 2026Updated 2 months ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- ☆59Jul 6, 2020Updated 5 years ago
- ☆15Apr 26, 2022Updated 3 years ago
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Mar 24, 2023Updated 3 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆135Sep 15, 2022Updated 3 years ago
- Enabling pure data parallel training of DLRM via caching and prefetching☆17Oct 29, 2021Updated 4 years ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆411Jun 14, 2025Updated 9 months ago
- [ICLR 2021] "UMEC: Unified Model and Embedding Compression for Efficient Recommendation Systems" by Jiayi Shen, Haotao Wang*, Shupeng Gui…☆39Dec 30, 2021Updated 4 years ago
- ☆28Oct 21, 2019Updated 6 years ago
- Pytorch domain library for recommendation systems☆2,499Updated this week
- BARS: Towards Open Benchmarking for Recommender Systems https://openbenchmark.github.io/BARS☆388Nov 8, 2024Updated last year
- A PyTorch implementation of ICLR 2021 paper: Learnable Embedding Sizes for Recommender Systems☆66Jan 21, 2021Updated 5 years ago
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,547Updated this week
- ☆11Oct 10, 2017Updated 8 years ago
- ☆18May 8, 2021Updated 4 years ago
- Metamodeling, sensitivity analysis and visualization using the tensor train format☆21Sep 8, 2022Updated 3 years ago
- This repo includes some graph-based CTR prediction models and other representative baselines.☆70Apr 23, 2024Updated last year
- A deep ranking personalization framework☆133Dec 17, 2025Updated 3 months ago
- ☆10Nov 28, 2020Updated 5 years ago
- ☆11Apr 5, 2021Updated 4 years ago
- PyTorch extensions for high performance and large scale training.☆3,404Apr 26, 2025Updated 10 months ago
- Chinese processing☆36Jan 29, 2014Updated 12 years ago
- Openfst mirror with some fixes☆15Aug 23, 2024Updated last year
- Implementation for the paper entitled "Session-based Recommendation with Self-Attention Networks"☆14Apr 29, 2021Updated 4 years ago
- Set of datasets for the deep learning recommendation model (DLRM).☆49Dec 21, 2022Updated 3 years ago
- Rax is a Learning-to-Rank library written in JAX.☆337Feb 13, 2026Updated last month
- AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Prediction☆184Dec 7, 2021Updated 4 years ago
- NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale da…☆1,139Mar 12, 2026Updated last week
- Probabilistic modelling with tensor networks☆29Nov 29, 2019Updated 6 years ago
- [Android Library] Make event and exception tracking like a piece of cake☆14Jun 13, 2020Updated 5 years ago
- PyTorch toolbox for matrix product state models☆165Mar 26, 2024Updated last year
- Tensor Network Learning with PyTorch☆306May 23, 2024Updated last year
- Successfully training approximations to full-rank matrices for efficiency in deep learning.☆17Jan 5, 2021Updated 5 years ago
- A library for romanization of widely-used languages using common romanization systems.☆14Feb 16, 2021Updated 5 years ago
- An implementation of RaFM. Xiaoshuang Chen, Yin Zheng, Jiaxiang Wang, et al. "RaFM: Rank-Aware Factorization Machines"☆12May 11, 2019Updated 6 years ago
- ☆41Feb 6, 2021Updated 5 years ago
- papers of universal user representation learning for recommendation☆25Mar 1, 2024Updated 2 years ago