facebookresearch / FBTT-EmbeddingLinks
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 …
☆194Updated 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
Sorting:
- Simple Distributed Deep Learning on TensorFlow☆133Updated last month
- Research and development for optimizing transformers☆129Updated 4 years ago
- High performance distributed framework for training deep learning recommendation models based on PyTorch.☆408Updated last month
- A tensor-aware point-to-point communication primitive for machine learning☆260Updated this week
- Slicing a PyTorch Tensor Into Parallel Shards☆299Updated 2 months ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆57Updated 2 years ago
- distributed-embeddings is a library for building large embedding based models in Tensorflow 2.☆45Updated last year
- Block-sparse primitives for PyTorch☆157Updated 4 years ago
- http://vlsiarch.eecs.harvard.edu/research/recommendation/☆136Updated 2 years ago
- A Tool for Automatic Parallelization of Deep Learning Training in Distributed Multi-GPU Environments.☆132Updated 3 years ago
- ☆251Updated last year
- A library for syntactically rewriting Python programs, pronounced (sinner).☆69Updated 3 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆132Updated 3 years ago
- PyTorch implementation of Hash Embeddings (NIPS 2017). Submission to the NIPS Implementation Challenge.☆199Updated 6 years ago
- Running BERT without Padding☆474Updated 3 years ago
- A GPU performance profiling tool for PyTorch models☆503Updated 4 years ago
- Training material for IPU users: tutorials, feature examples, simple applications☆86Updated 2 years ago
- PyTorch RFCs (experimental)☆134Updated 2 months ago
- DLPack for Tensorflow☆35Updated 5 years ago
- Time-based Sequence Model for Personalization and Recommendation Systems☆49Updated 3 years ago
- Stride visualizations☆38Updated 7 years ago
- Development repository for integrating FlexFlow (A distributed deep learning framework that supports flexible parallelization strategies)…☆28Updated 3 years ago
- Torch Distributed Experimental☆117Updated last year
- Implementation of a Transformer, but completely in Triton☆273Updated 3 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆208Updated 2 years ago
- Distributed ML Optimizer☆32Updated 4 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆158Updated last month
- Practical low-rank gradient compression for distributed optimization: https://arxiv.org/abs/1905.13727☆147Updated 9 months ago
- Guide for building custom op for TensorFlow☆382Updated 2 years ago