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:
- 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,060Jan 12, 2026Updated 6 months ago
- ☆59Jul 6, 2020Updated 6 years ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- ☆15Apr 26, 2022Updated 4 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.
- [KDD 2022] AutoShard: Automated Embedding Table Sharding for Recommender Systems☆22Mar 24, 2023Updated 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.☆414Jul 20, 2026Updated last week
- [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,588Updated 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
- 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.
- 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
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/☆1,572Updated this week
- ☆11Oct 10, 2017Updated 8 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
- Metamodeling, sensitivity analysis and visualization using the tensor train format☆21Sep 8, 2022Updated 3 years ago
- ☆10Nov 28, 2020Updated 5 years ago
- A deep ranking personalization framework☆135Dec 17, 2025Updated 7 months ago
- This repo includes some graph-based CTR prediction models and other representative baselines.☆72Apr 23, 2024Updated 2 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.
- ☆11Apr 5, 2021Updated 5 years ago
- PyTorch extensions for high performance and large scale training.☆3,411Apr 26, 2025Updated last year
- Implementation for the paper entitled "Session-based Recommendation with Self-Attention Networks"☆14Apr 29, 2021Updated 5 years ago
- AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Prediction☆184Dec 7, 2021Updated 4 years ago
- Rax is a Learning-to-Rank library written in JAX.☆339Jun 23, 2026Updated last month
- NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale da…☆1,149May 22, 2026Updated 2 months ago
- Set of datasets for the deep learning recommendation model (DLRM).☆51Dec 21, 2022Updated 3 years ago
- ☆27Aug 19, 2022Updated 3 years ago
- Probabilistic modelling with tensor networks☆29Nov 29, 2019Updated 6 years ago
- 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.
- 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
- 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
- [ICLR 2022] Code for paper "Exploring Extreme Parameter Compression for Pre-trained Language Models"(https://arxiv.org/abs/2205.10036)☆22May 24, 2023Updated 3 years ago
- An implementation of RaFM. Xiaoshuang Chen, Yin Zheng, Jiaxiang Wang, et al. "RaFM: Rank-Aware Factorization Machines"☆12May 11, 2019Updated 7 years ago