LLM-Pruning Collection v1.0
Open Source
ML
Infrastructure
LLM-Pruning Collection is a JAX-based repository that consolidates major pruning algorithms for large language models into a single, reproducible framework. It aims to facilitate easy comparison of block-level, layer-level, and weight-level pruning methods under a consistent training and evaluation stack on both GPUs and TPUs.
Consolidates various pruning algorithms for large language models
Supports block-level, layer-level, and weight-level pruning methods
Compatible with both GPUs and TPUs for training and evaluation