mxnet-memonger

Memory optimizer

A tool for optimizing deep learning models to reduce memory usage without sacrificing performance

Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets

GitHub

308 stars
19 watching
63 forks
Language: Python
last commit: over 7 years ago

Related projects:

Repository Description Stars
mapillary/inplace_abn An optimization technique to reduce memory usage in deep neural networks during training 1,321
fmassa/optimize-net An optimization library for reducing memory usage in PyTorch neural networks 282
dmlc/mxnet.js A JavaScript package for running deep learning models in the browser without requiring a server 435
neuralmagic/sparseml Enables the creation of smaller neural network models through efficient pruning and quantization techniques 2,071
brml/climin A framework for optimizing machine learning functions using gradient-based optimization methods. 180
xternalz/sdpoint A deep learning method for optimizing convolutional neural networks by reducing computational cost while improving regularization and inference efficiency. 18
mingruiliu-ml-lab/episode An algorithm for Federated Learning with heterogeneous data, designed to optimize deep neural networks and improve performance 2
typemonkey/turtle A programming language designed to optimize memory usage through long-term storage of data 5
dmlc/mxnet-gtc-tutorial An interactive tutorial on deep learning with MXNet using Jupyter Notebooks 131
mlpack/ensmallen A high-quality C++ library for non-linear numerical optimization 743
toddsundsted/mxnet.cr Provides bindings for MXNet, an open source deep learning framework written in C++ 22
davisyoshida/lorax A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. 132
tdr-autosync/mi-lib-django_memoize Caches and optimizes expensive function calls in Django applications. 120
bruinxiong/modified-crunet-and-residual-attention-network.mxnet An MXNet implementation of a modified deep neural network architecture for image classification 67
deepakkumar1984/mxnet.sharp A .NET Standard library providing C# bindings for the Apache MXNet deep learning framework 149