PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
tensorflow
tensorflow-model-optimization

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

105K 2K 347
horseee
deepcache

[CVPR 2024] DeepCache: Accelerating Diffusion Models for Free

62K 963 52
Microsoft
nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

34K 14K 2K
VainF
torch-pruning

[CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.

23K 3K 378
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
kxytechnologies
kxy

A Powerful Serverless Pre-Learning and Post-Learning Analysis Toolkit

4K 51 12
666DZY666
micronet

A model compression and deploy lib.

2K 2K 477
FasterAI-Labs
fasterai

FasterAI: Prune and Distill your models with FastAI and PyTorch

1K 261 19
GeoffreyWang1117
uni-layer

30+ layer contribution metrics from 7 theoretical categories for PyTorch model compression. Bridges for Torch-Pruning and PEFT/LoRA.

996 0 0
Picovoice
picollm

On-device LLM Inference Powered by X-Bit Quantization

980 311 25
r-papso
torch-optim

PyTorch models optimization by neural network pruning

812 3 1
Microsoft
nni-daily

Neural Network Intelligence package

798 14K 2K
aquvitae
aquvitae

The easiest Knowledge Distillation library for Light Weight DeepLearning

629 88 10
danhicks96
prismkv

3-D Stacked-Plane KV Cache Quantizer — defensive prior art publication

575 1 1
gershonc
octopus-ml

A collection of handy ML and data visualization and validation tools. Go ahead and train, evaluate and validate your ML models and data with minimal effort.

572 23 5
lpalbou
model-quantizer

A tool for quantizing large language models

530 2 0
tianyic
only-train-once

Only Train Once (OTO): Automatic One-Shot General DNN Training and Compression Framework

511 311 48
SforAiDL
kd-lib

A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quantization.

450 649 61
Picovoice
picollmdemo

picoLLM Inference Engine demos

424 311 25
musco-ai
musco-pytorch

MUSCO: Multi-Stage COmpression of Neural Networks

342 72 17
mlzxy
qsparse

Train neural networks with joint quantization and pruning on both weights and activations using any pytorch modules

313 42 2
microsoft
archai

Platform for Neural Architecture Search

241 486 93
Argonaut790
fused-turboquant

Fused Triton encode/decode kernels for TurboQuant KV cache compression, powered by Randomized Hadamard Transform.

208 8 0
m-pektas
bfas

Brute Force Architecture Search

183 4 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery