Rachel Gordon | MIT CSAIL
View original ↗Implement a control-theory based pruning callback for PyTorch or JAX trainers. This would enable dynamic model shrinking during the training loop without full retraining cycles.
Suggested repo: ctrl-prune
"Shrink your models during training using control theory."
Estimated effort: 80h