llm-pruning-collection
Collection of LLM pruning methods and training code for GPUs & TPUs.
GraphCanon updated Sep 9, 2026 · GitHub synced Sep 9, 2026
32views this month
Decision brief
The llm-pruning-collection repository provides a comprehensive set of large language model pruning methods, along with the necessary training and evaluation scripts for GPUs and TPUs.
Good fit when
- When you are working on reducing the size or improving inference speed of large language models using various pruning techniques available in this collection.
- If you need specific fine-tuning and pretraining framework options for both GPU and TPU platforms, which support a variety of popular pruning methods and their implementations.
Avoid when
- Avoid if your project requires a pruning method that is not included in the collection or if the current platform capabilities do not align with your hardware requirements.
- Not suitable for those who need tools to train models from scratch rather than focusing on model pruning and optimization techniques.
- Pricing:
- freemium - The software is free and open-source, licensed under Apache-2.0, but users must provide their own hardware or use cloud services like Google TPU Research Cloud for computational resources.
- Requirements:
- The repository includes pretraining and fine-tuning scripts for both GPU and TPU platforms.; A JAX-based environment is required to run the code in this repository.
Observed Jul 16, 2026 · Source: enrich:decision_facts
Verify the decision
Maintenance and security
Full trust report- Maintenance
- Slowing (141d since push)
- As of Sep 9, 2026
- Provenance
- Not a fork · Organization account
- As of Sep 9, 2026
- Security (OSV)
- No lockfile
- As of Jul 15, 2026
Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.
Install
pip install llm-pruning-collectionSimilar tools
Same-category neighbours. No typed graph edges are catalogued for this tool yet.
Evidence and technical details
Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.
Overview
Includes various implementations of LLM pruning, pretraining and fine-tuning codes, and evaluation scripts for GPU and TPU platforms.
Capability facts
- Languages
- python
Source: github.language · Sep 9, 2026
Categories
Tags
README
Small LLMs: Pruning vs Training from Scratch Yufeng Xu<sup 1</sup , Taiming Lu<sup 1</sup , Jiachen Zhu<sup 2</sup , Mingjie Sun<sup 3</sup , Kunjun Li<sup 1</sup , and Zhuang Liu<sup 1</sup 1 Princeton. 2 NYU. 3 CMU. This is a Jax based repo for LLM Prunning, It contains: the im...
For agents
This page has a .md twin and JSON over the API.