waitwaitforget / modelparallel_pytorch
Model Parallelism for pytorch training multiple network on multiple GPUs.
☆28Updated 6 years ago
Alternatives and similar repositories for modelparallel_pytorch:
Users that are interested in modelparallel_pytorch are comparing it to the libraries listed below
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- This repository is no longer maintained. Check☆81Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated last month
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆44Updated 4 years ago
- NeurIPS'19: Meta-Weight-Net: Learning an Explicit Mapping For Sample Weighting (Pytorch implementation for class imbalance).☆33Updated 5 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- ☆47Updated 4 years ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated last year
- A PyTorch implementation of Conditional PixelCNNs☆27Updated 7 years ago
- An implementation of shampoo☆74Updated 6 years ago
- A template for PyTorch projects.☆22Updated 6 years ago
- Notes and tutorials on "Mutual information and self-supervised learning"☆39Updated 5 years ago
- PyTorch Examples repo for "ReZero is All You Need: Fast Convergence at Large Depth"☆62Updated 6 months ago
- Implementation of the reversible residual network in pytorch☆102Updated 3 years ago
- [ICLR 2019] ProbGAN: Towards Probabilistic GAN with Theoretical Guarantees☆32Updated 5 years ago
- PyTorch Implementations of Dropout Variants☆87Updated 7 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 4 years ago
- ☆23Updated 6 years ago
- Net2Net implementation on PyTorch for any possible vision layers.☆38Updated 7 years ago
- AdaX: Adaptive Gradient Descent with Exponential Long Term Momery☆34Updated 4 years ago
- ☆27Updated 4 years ago
- ☆62Updated 4 years ago
- Reference implementation of the PAL optimizer☆20Updated 4 years ago
- PyTorch implementation of HashedNets☆36Updated last year
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 5 years ago
- Exploiting Uncertainty of Loss Landscape for Stochastic Optimization☆15Updated 5 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 4 years ago
- AdaBatch: Adaptive Batch Sizes for Training Deep Neural Networks☆42Updated 7 years ago