mtbrandy / pytorch
Please visit https://github.com/IBM/pytorch-large-model-support for the latest information on PyTorch LMS.
☆21Updated 4 years ago
Alternatives and similar repositories for pytorch:
Users that are interested in pytorch are comparing it to the libraries listed below
- Large Model Support in PyTorch☆133Updated 3 years ago
- To train deep convolutional neural networks, the input data and the activations need to be kept in memory. Given the limited memory avail…☆100Updated last year
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- This library is designed to speed up huge model training on unified memory.☆28Updated 3 years ago
- Framework for creating (partially) reversible neural networks with PyTorch☆152Updated 2 years ago
- This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".☆41Updated last year
- Scheduling GPU cluster workloads with Slurm☆74Updated 6 years ago
- Training with FP16 weights in PyTorch☆77Updated 5 years ago
- ☆165Updated 6 years ago
- Large Model Support in Tensorflow☆202Updated 4 years ago
- ☆174Updated 3 years ago
- Python bindings for NVTX☆66Updated last year
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- Unofficial implementation of "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" (https://arxiv.org/abs/2103.14030)☆65Updated 3 years ago
- This Python package enables the training and inference of deep learning models for very large data, such as megapixel images, using atten…☆98Updated 5 years ago
- Convert nvprof profiles into about:tracing compatible JSON files☆69Updated 4 years ago
- Repository containing code for "U-Net Fixed-Point Quantization for Medical Image Segmentation" paper at MICCAI2019☆85Updated 2 years ago
- Hyppopy is a python toolbox for blackbox optimization. It's purpose is to offer a unified and easy to use interface to a collection of so…☆27Updated 3 years ago
- Self-supervised learning for microscopy images☆55Updated 5 years ago
- Unofficial PyTorch Implementation of EvoNorm☆121Updated 3 years ago
- NGC Container Replicator☆28Updated 2 years ago
- Efficient Segmentation for Volumetric Data☆44Updated 6 years ago
- A GPipe implementation in PyTorch☆836Updated 9 months ago
- Code for "How to Initialize your Network? Robust Initialization for WeightNorm & ResNets"☆14Updated last year
- Material for the SC21 Deep Learning at Scale Tutorial☆25Updated 2 years ago
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆197Updated 5 years ago
- ☆69Updated 5 years ago
- Lightweight and Parallel Deep Learning Framework☆261Updated 2 years ago
- PyTorch implementations of `BatchSampler` that under/over sample according to a chosen parameter alpha, in order to create a balanced tra…☆84Updated 5 years ago
- Use streaming to train whole-slides images with single image-level labels, by reducing GPU memory requirements with 99%.☆79Updated 2 years ago