optimize-net
Memory Optimizer
An optimization library for reducing memory usage in PyTorch neural networks
OptNet - Reducing memory usage in torch neural nets
282 stars
23 watching
45 forks
Language: Lua
last commit: over 9 years agoLinked from 1 awesome list
carpedm20/awesome-torchRelated projects:
| Repository | Description | Stars |
|---|---|---|
 mapillary/inplace_abn | An optimization technique to reduce memory usage in deep neural networks during training | 1,323 |
 dmlc/mxnet-memonger | A tool for optimizing deep learning models to reduce memory usage without sacrificing performance | 308 |
 anjiecheng/mnasnet-pytorch | An implementation of a neural network architecture search algorithm optimized for mobile devices. | 271 |
 mc-nya/fednest | An implementation of a federated optimization algorithm for distributed machine learning | 6 |
 rentruewang/koila | A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution | 1,823 |
 gpleiss/efficient_densenet_pytorch | An implementation of DenseNets optimized to save GPU memory by using checkpointing | 1,524 |
 atgambardella/pytorch-es | An implementation of an optimization algorithm for training neural networks in machine learning environments. | 351 |
 dylanmuir/fmin_adam | An implementation of the Adam optimisation algorithm for stochastic gradient descent problems in Matlab. | 56 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 134 |
 cma-es/pycma | An optimization algorithm implementation in Python | 1,123 |
 typemonkey/turtle | A programming language designed to optimize memory usage through long-term storage of data | 5 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 517 |
 mengomarlene/opti4abq | An optimisation method that minimises the difference between FEA output and data in Abaqus models | 17 |
 neuralmagic/sparseml | Enables the creation of smaller neural network models through efficient pruning and quantization techniques | 2,083 |
 andreaferretti/memo | A package that provides functions to memoize (cache) and optimize Nim functions. | 78 |