open-mmlab/mmrazor
A PyTorch-based toolbox for compressing deep learning models through pruning, quantization, knowledge distillation, and neural architecture search.

This repository is part of the OpenMMLab ecosystem and provides a comprehensive model compression toolkit. It supports multiple compression techniques including network pruning, quantization, knowledge distillation, and neural architecture search (NAS). Built on PyTorch, it offers tools for compressing models across computer vision tasks like classification, detection, and segmentation, with support for methods such as AutoSlim and DARTS.