pytorch / elasticLinks
PyTorch elastic training
☆729Updated 3 years ago
Alternatives and similar repositories for elastic
Users that are interested in elastic are comparing it to the libraries listed below
Sorting:
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Updated 5 years ago
- A GPU performance profiling tool for PyTorch models☆509Updated 4 years ago
- A GPipe implementation in PyTorch☆857Updated last year
- Fast Block Sparse Matrices for Pytorch☆550Updated 4 years ago
- A uniform interface to run deep learning models from multiple frameworks☆941Updated last year
- Mesh TensorFlow: Model Parallelism Made Easier☆1,624Updated 2 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,061Updated 2 years ago
- Collective communications library with various primitives for multi-machine training.☆1,370Updated last week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆402Updated this week
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 5 months ago
- Bagua Speeds up PyTorch☆881Updated last year
- High performance model preprocessing library on PyTorch☆646Updated last year
- Distributed Computing for AI Made Simple☆1,047Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆275Updated 3 weeks ago
- A performant and modular runtime for TensorFlow☆757Updated 2 months ago
- DAWNBench: An End-to-End Deep Learning Benchmark and Competition☆263Updated 5 years ago
- PyTorch layer-by-layer model profiler☆608Updated 4 years ago
- Pytorch Lightning Distributed Accelerators using Ray☆215Updated 2 years ago
- Providing reproducibility in deep learning frameworks☆432Updated last year
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆734Updated last week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,067Updated last year
- Guide for building custom op for TensorFlow☆385Updated 2 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- Python 3 Bindings for NVML library. Get NVIDIA GPU status inside your program.☆249Updated 3 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆998Updated this week
- TVM integration into PyTorch☆455Updated 5 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆699Updated 2 years ago
- A multi-model machine learning feature embedding database☆639Updated 5 years ago
- ☆306Updated 4 years ago
- ☆392Updated 3 years ago