cybertronai / gradient-checkpointingLinks
Make huge neural nets fit in memory
☆2,803Updated 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:
- Mesh TensorFlow: Model Parallelism Made Easier☆1,613Updated last year
- PyTorch extensions for high performance and large scale training.☆3,350Updated 3 months ago
- A lightweight library for PyTorch training tools and utilities☆1,700Updated last week
- ☆1,208Updated 5 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,044Updated 2 years ago
- A Python module for getting the GPU status from NVIDA GPUs using nvidia-smi programmically in Python☆1,196Updated last year
- Profiling and inspecting memory in pytorch☆1,065Updated 11 months ago
- Fast & Simple Resource-Constrained Learning of Deep Network Structure