koila
by rentruewang
Prevent PyTorch's CUDA error: out of memory in just 1 line of code.
AI summary
Memory optimizer for deep learning models
A lightweight wrapper around PyTorch to prevent CUDA out-of-memory errors and optimize model execution
- stars
- 1.8K
- forks
- 63
- watching
- 12
Similar projects
Found by comparing what the projects do, not just their names.
Deep learning toolkit
A Python framework for building deep learning models with optimized encoding layers and batch normalization.
RVAE
A deep learning implementation of a recurrent variational autoencoder for generating sequential data.
Reinforcement Learning framework
Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks
Deep learning models
Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets.
Meta learner
A PyTorch implementation of meta-learning using gradient descent to adapt to new tasks.
Memory optimizer
A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation.
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Optimization algorithm
A PyTorch implementation of an optimization algorithm for continuous control and reinforcement learning tasks
Deep Learning Wrapper
Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages.
Message passing abstraction
Decoupled Neural Interfaces using Synthetic Gradients for PyTorch
Neural network optimizer
An implementation of an optimization algorithm for training neural networks in machine learning environments.
Deep learning accelerator
A framework for accelerating PyTorch deep learning training
Batching library
A PyTorch module for dynamic batching and optimized computation on deep neural networks
Pruning tool
This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference.
Neural net trainer
A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities
