pytorch / elasticLinks
PyTorch elastic training
☆728Updated 3 years ago
Alternatives and similar repositories for elastic
Users that are interested in elastic are comparing it to the libraries listed below
Sorting:
- A GPU performance profiling tool for PyTorch models☆503Updated 3 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- A GPipe implementation in PyTorch☆843Updated 10 months ago
- High performance model preprocessing library on PyTorch☆650Updated last year
- PyTorch layer-by-layer model profiler☆607Updated 4 years ago
- A uniform interface to run deep learning models from multiple frameworks☆934Updated last year
- Fast Block Sparse Matrices for Pytorch☆547Updated 4 years ago
- Collective communications library with various primitives for multi-machine training.☆1,315Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆367Updated this week
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,041Updated 2 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆696Updated 2 years ago
- Mesh TensorFlow: Model Parallelism Made Easier☆1,608Updated last year
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- A performant and modular runtime for TensorFlow☆762Updated 2 months ago
- Implementations of ideas from recent papers☆393Updated 4 years ago
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆194Updated 2 years ago
- Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet f…☆1,840Updated last year
- functorch is JAX-like composable function transforms for PyTorch.☆1,432Updated this week
- A tensor-aware point-to-point communication primitive for machine learning☆258Updated 2 years ago
- Distributed Computing for AI Made Simple☆1,043Updated 2 years ago
- DAWNBench: An End-to-End Deep Learning Benchmark and Competition☆262Updated 4 years ago
- Bagua Speeds up PyTorch☆883Updated 10 months ago
- PyTorch functions and utilities to make your life easier☆195Updated 4 years ago
- common in-memory tensor structure☆1,014Updated last week
- Slicing a PyTorch Tensor Into Parallel Shards☆299Updated 2 weeks ago
- TVM integration into PyTorch☆453Updated 5 years ago
- Named Tensor implementation for Torch☆444Updated 2 years ago
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆948Updated 2 weeks ago
- torchbearer: A model fitting library for PyTorch☆640Updated last year
- A multi-model machine learning feature embedding database☆638Updated 5 years ago