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 …
☆193Jul 20, 2022Updated 4 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:
- An implementation of a deep learning recommendation model (DLRM)☆4,066Jan 12, 2026Updated 7 months ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training☆1,071Aug 3, 2026Updated 2 weeks ago
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Mar 24, 2023Updated 3 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆134Sep 15, 2022Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- [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
- Tutorials about AutoML☆87Feb 24, 2022Updated 4 years ago
- Pytorch domain library for recommendation systems☆2,599Updated this week
- Time-based Sequence Model for Personalization and Recommendation Systems☆50Aug 26, 2021Updated 4 years ago
- BARS: Towards Open Benchmarking for Recommender Systems https://openbenchmark.github.io/BARS☆392Nov 8, 2024Updated last year
- A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…☆18Jan 5, 2023Updated 3 years ago
- A PyTorch implementation of ICLR 2021 paper: Learnable Embedding Sizes for Recommender Systems☆66Jan 21, 2021Updated 5 years ago
- ☆11Oct 10, 2017Updated 8 years ago
- 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.
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Jul 2, 2020Updated 6 years ago
- UBR4CTR is the code for our proposed User Behavior Retrieval for CTR Prediction framework in SIGIR 2020.☆81Jul 14, 2021Updated 5 years ago
- A deep ranking personalization framework☆135Dec 17, 2025Updated 8 months ago
- This repo includes some graph-based CTR prediction models and other representative baselines.☆72Apr 23, 2024Updated 2 years ago
- PyTorch extensions for high performance and large scale training.☆3,408Apr 26, 2025Updated last year
- Implementation for the paper entitled "Session-based Recommendation with Self-Attention Networks"☆14Apr 29, 2021Updated 5 years ago
- Rax is a Learning-to-Rank library written in JAX.☆341Aug 6, 2026Updated last week
- Performs a faster tensor train (TT) decomposition for large sparse data☆14Sep 7, 2020Updated 5 years ago
- NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale da…☆1,151May 22, 2026Updated 2 months 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.
- Probabilistic modelling with tensor networks☆29Nov 29, 2019Updated 6 years ago
- FINN.no Slate Dataset for Recommender Systems. A dataset containing all interactions (viewed items + response (clicked item / no click) f…☆54Jan 29, 2023Updated 3 years ago
- PyTorch toolbox for matrix product state models☆165Mar 26, 2024Updated 2 years ago
- Tensor Network Learning with PyTorch☆310May 23, 2024Updated 2 years ago
- Successfully training approximations to full-rank matrices for efficiency in deep learning.☆16Jan 5, 2021Updated 5 years ago
- ☆41Feb 6, 2021Updated 5 years ago
- An offline evaluation framework for sequence-based recommender systems☆13May 17, 2019Updated 7 years ago
- papers of universal user representation learning for recommendation☆25Mar 1, 2024Updated 2 years ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆66Feb 22, 2022Updated 4 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.
- An implementation of RaFM. Xiaoshuang Chen, Yin Zheng, Jiaxiang Wang, et al. "RaFM: Rank-Aware Factorization Machines"☆12May 11, 2019Updated 7 years ago
- typesafe implementation of tensor algebra in Haskell☆13Feb 27, 2021Updated 5 years ago
- An implementation of various tensor-based decomposition for NN & RNN parameters☆18Jun 4, 2018Updated 8 years ago
- pytorch implementation of wavenet autoencoder https://arxiv.org/pdf/1704.01279.pdf☆12Jul 25, 2018Updated 8 years ago
- Deep convolutional tensor network☆11Sep 29, 2020Updated 5 years ago
- High performance model preprocessing library on PyTorch☆641Mar 29, 2024Updated 2 years ago
- Python code for implementing embeddings in the Wasserstein space of elliptical distributions☆11Jul 22, 2020Updated 6 years ago