pytorch-memory-optim
Python
★ 93
updated 2y ago
This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog post.
No plain-English explanation yet — one is being written right now. Check back in a minute.