PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Sparsity Python Packages

Python packages with the GitHub topic sparsity. Sorted by relevance, with stars and monthly downloads.
pytorch
torchao

PyTorch native quantization and sparsity for training and inference

3.8M 3K 505
openvinotoolkit
nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

482K 1K 294
tensorflow
tensorflow-model-optimization

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

111K 2K 347
intel
neural-compressor

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

32K 3K 305
tensorflow
tf-model-optimization-nightly

A suite of tools that users, both novice and advanced can use to optimize machine learning models for deployment and execution.

10K 2K 347
neuralmagic
sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models

3K 2K 156
bwohlberg
sporco

Sparse Optimisation Research Code

2K 273 40
intel
neural-compressor-pt

Repository of Intel® Neural Compressor

1K 3K 305
adrhill
asdex

Automatic Sparse Differentiation in JAX.

1K 25 3
satabios
sconce

E2E AutoML Model Compression Package

1K 45 4
intel
neural-compressor-tf

Repository of Intel® Neural Compressor

1K 3K 305
FasterAI-Labs
fasterai

FasterAI: Prune and Distill your models with FastAI and PyTorch

1K 261 19
intel
neural-speed

Repository of Intel® Intel Extension for Transformers

976 353 38
intel
neural-compressor-full

Repository of Intel® Neural Compressor

464 3K 305
intel
neural-solution

Repository of Intel® Neural Compressor

423 3K 305
intel
neural-insights

Repository of Intel® Neural Compressor

355 3K 305
intel
lpot

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

285 3K 306
SIP-AAU
magni

A package for AFM image reconstruction and compressed sensing in general

167 43 10
RabadanLab
randomly

A Library for Denoising Single-Cell Data with Random Matrix Theory

137 38 10
danielathome19
tensorflow-engram

tensorflow-engram: A Python package for Engram Neural Networks, adding biologically-inspired Hebbian memory and engram layers to TensorFlow/Keras models, supporting memory traces, plasticity, attention, and sparsity for neural sequence learning.

124 4 1
intel
ilit

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

105 3K 306
fraenkel-lab
gslr

An algorithm for classification from a graph-sparse support

75 15 4
ryantd
veloce

WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.

69 17 0
intel
neural-compressor-3x-pt

SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, and ONNX Runtime

16 3K 305
    • Data from PyPI, GitHub, ClickHouse, and BigQuery