waitwaitforget / modelparallel_pytorch
Model Parallelism for pytorch training multiple network on multiple GPUs.
☆28Updated 7 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
- An implementation of shampoo☆74Updated 7 years ago
- ☆47Updated 4 years ago
- (Batched) advanced indexing for PyTorch.☆53Updated 3 months ago
- AdaBatch: Adaptive Batch Sizes for Training Deep Neural Networks☆42Updated 7 years ago
- tunz's CUDA pytorch operator (MaskedSoftmax)☆75Updated 6 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated last year
- ☆54Updated 3 years ago
- an implementation of L0 regularization with PyTorch☆57Updated 6 years ago
- Non official pytorch implementation of i-Resnet, invertible residual networks.☆25Updated 3 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆53Updated 5 years ago
- PyTorch DataLoader processed in multiple remote computation machines for heavy data processings☆67Updated 5 years ago
- Code for Variational Laplace Autoencoders☆55Updated last year
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- Implementation of the reversible residual network in pytorch☆104Updated 3 years ago
- Exploiting Uncertainty of Loss Landscape for Stochastic Optimization☆15Updated 5 years ago
- CCAs for looking into DNNs☆70Updated 4 years ago
- Official PyTorch implementation of Harmonizing Maximum Likelihood with GANs for Multimodal Conditional Generation (ICLR 2019)☆93Updated 8 months ago
- NeurIPS'19: Meta-Weight-Net: Learning an Explicit Mapping For Sample Weighting (Pytorch implementation for class imbalance).☆33Updated 5 years ago
- ☆26Updated 5 years ago
- Delta Orthogonal Initialization for PyTorch☆18Updated 6 years ago
- Unofficial pytorch implementation of ReZero in ResNet☆23Updated 4 years ago
- Unofficial implementation of Stand-Alone Self-Attention in Vision Models (obsolete)☆44Updated 5 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 5 years ago
- Pytorch implementation of Group Equivariant Capsule Networks☆29Updated 6 years ago
- Distributed, mixed-precision training with PyTorch☆89Updated 4 years ago
- Fork of diux-dev/imagenet18☆15Updated 6 years ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆45Updated 5 years ago
- Net2Net implementation on PyTorch for any possible vision layers.☆38Updated 7 years ago