pytorch / elastic
PyTorch elastic training
☆730Updated 2 years ago
Alternatives and similar repositories for elastic:
Users that are interested in elastic are comparing it to the libraries listed below
- A GPU performance profiling tool for PyTorch models☆504Updated 3 years ago
- A uniform interface to run deep learning models from multiple frameworks☆936Updated last year
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- A GPipe implementation in PyTorch☆833Updated 7 months ago
- PyTorch layer-by-layer model profiler☆606Updated 3 years ago
- PyTorch on Kubernetes☆308Updated 3 years ago
- High performance model preprocessing library on PyTorch☆651Updated 11 months ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆349Updated this week
- TVM integration into PyTorch☆452Updated 5 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- A performant and modular runtime for TensorFlow☆759Updated 2 weeks ago
- ☆109Updated 3 years ago
- Mesh TensorFlow: Model Parallelism Made Easier☆1,603Updated last year
- Experimental ground for optimizing memory of pytorch models☆365Updated 6 years ago
- A multi-model machine learning feature embedding database☆637Updated 5 years ago
- Automated Machine Learning on Kubernetes☆1,545Updated 2 weeks ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,038Updated last year
- Implementations of ideas from recent papers☆392Updated 4 years ago
- Bagua Speeds up PyTorch☆878Updated 7 months ago
- Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet f…☆1,821Updated last year
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆696Updated last year
- FastFormers - highly efficient transformer models for NLU☆704Updated last year
- Facebook AI Performance Evaluation Platform☆390Updated last month
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆193Updated 2 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- Distributed Computing for AI Made Simple☆1,042Updated last year
- DAWNBench: An End-to-End Deep Learning Benchmark and Competition☆260Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆372Updated 4 years ago