simtony / tf_multigpu
Multi-gpu/distributed training script in Tensorflow 1.x.
☆17Updated 4 years ago
Related projects: ⓘ
- OneFlow models for benchmarking.☆104Updated last month
- A Fast Muti-processing BERT-Inference System☆100Updated last year
- A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch☆79Updated 4 years ago
- alibabacloud-aiacc-demo☆42Updated last year
- pytorch源码阅读 0.2.0 版本☆88Updated 4 years ago
- Simple Dynamic Batching Inference☆145Updated 2 years ago
- Tutorial code on how to build your own Deep Learning System in 2k Lines☆126Updated 7 years ago
- oneflow documentation☆68Updated 2 months ago
- Models and examples built with OneFlow☆94Updated last month
- PyTorch Dataset Rank Dataset☆38Updated 3 years ago
- A universal and efficient framework for training well-performing light net☆124Updated 7 years ago
- A small deep-learning framework with C++/Python/CUDA☆53Updated 6 years ago
- ☆50Updated 11 months ago
- InsNet Runs Instance-dependent Neural Networks with Padding-free Dynamic Batching.☆66Updated 2 years ago
- an automatic differentiation framework with dynamic graph/支持动态图的自动求导框架☆101Updated 4 years ago
- A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster☆152Updated 5 months ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆129Updated last year
- Python Scritpt which can be embedded into PyTorch model to print the model size.☆18Updated 3 years ago
- ☆169Updated 3 years ago
- Analyze TensorFlow source code☆19Updated 7 years ago
- Transformer related optimization, including BERT, GPT☆58Updated last year
- Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.☆260Updated last year
- ☆205Updated last year
- pytorch读取tfrecords,构造数据流☆16Updated 5 years ago
- A flexible, high-performance framework for large-scale retrieval problems based on TensorFlow.☆147Updated 2 months ago
- Large batch training of CTR models based on DeepCTR with CowClip.☆162Updated last year
- PyTorch On Angel, arming PyTorch with a powerful Parameter Server, which enable PyTorch to train very big models.☆164Updated last year
- Distributed DataLoader For Pytorch Based On Ray☆24Updated 2 years ago
- [ICLR 2021] "UMEC: Unified Model and Embedding Compression for Efficient Recommendation Systems" by Jiayi Shen, Haotao Wang*, Shupeng Gui…☆39Updated 2 years ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago