cybertronai / gradient-checkpointingLinks
Make huge neural nets fit in memory
☆2,814Updated 5 years ago
Alternatives and similar repositories for gradient-checkpointing
Users that are interested in gradient-checkpointing are comparing it to the libraries listed below
Sorting:
- PyTorch extensions for high performance and large scale training.☆3,376Updated 5 months ago
- A lightweight library for PyTorch training tools and utilities☆1,713Updated this week
- Mesh TensorFlow: Model Parallelism Made Easier☆1,615Updated last year
- Enabling PyTorch on XLA Devices (e.g. Google TPU)☆2,688Updated this week
- Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"☆1,588Updated 5 years ago
- Experimental ground for optimizing memory of pytorch models☆367Updated 7 years ago
- Python library to easily log experiments and parallelize hyperparameter search for neural networks☆736Updated 3 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,053Updated 2 years ago
- Profiling and inspecting memory in pytorch☆1,071Updated last month
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆685Updated 5 years ago
- Train AI models efficiently on medical images using any framework☆1,874Updated last year
- ☆1,207Updated 5 years ago
- PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"☆2,724Updated 2 years ago
- A GPipe implementation in PyTorch☆855Updated last year
- torchbearer: A model fitting library for PyTorch☆642Updated last year
- A domain specific language to express machine learning workloads.☆1,760Updated 2 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,701Updated last week
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,852Updated last year
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆989Updated last week
- Fast & Simple Resource-Constrained Learning of Deep Network Structure☆1,033Updated 3 months ago
- Demo of running NNs across different frameworks☆1,655Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,141Updated last year
- PyTorch elastic training☆730Updated 3 years ago
- Providing reproducibility in deep learning frameworks☆428Updated last year
- Gin provides a lightweight configuration framework for Python☆2,126Updated last week
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,115Updated 3 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,550Updated 4 years ago
- Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.☆4,336Updated 3 weeks ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,061Updated last year
- TensorFlow Code for paper "Efficient Neural Architecture Search via Parameter Sharing"☆1,581Updated 6 years ago