Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
☆2,142Jun 2, 2025Updated 9 months ago
Alternatives and similar repositories for sparseml
Users that are interested in sparseml are comparing it to the libraries listed below
Sorting:
- Sparsity-aware deep learning inference runtime for CPUs☆3,163Jun 2, 2025Updated 9 months ago
- Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes☆387Jun 2, 2025Updated 9 months ago
- Top-level directory for documentation and general content☆120Jun 2, 2025Updated 9 months ago
- ML model optimization product to accelerate inference.☆326Jun 2, 2025Updated 9 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆267Dec 4, 2025Updated 3 months ago
- SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, …☆2,598Updated this week
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆877Aug 20, 2024Updated last year
- A data augmentations library for audio, image, text, and video.☆5,070Mar 14, 2026Updated last week
- Accessible large language models via k-bit quantization for PyTorch.☆8,052Updated this week
- RL research on Android devices.☆1,196Feb 26, 2026Updated 3 weeks ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,926Mar 10, 2026Updated last week
- 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization…☆3,332Mar 13, 2026Updated last week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,986Jun 16, 2024Updated last year
- the GPU-native, sandboxed Postgres for AI agents☆9,040Feb 16, 2026Updated last month
- Development repository for the Triton language and compiler☆18,708Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,739Updated this week
- 🐍 Geometric Computer Vision Library for Spatial AI☆11,121Updated this week
- Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀☆1,687Oct 23, 2024Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,709Updated this week
- [CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.☆3,267Sep 7, 2025Updated 6 months ago
- AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.☆2,566Mar 14, 2026Updated last week
- Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains☆1,731Oct 8, 2023Updated 2 years ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,869Updated this week
- Transformers-compatible library for applying various compression algorithms to LLMs for optimized deployment with vLLM☆2,891Updated this week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,046Jan 23, 2026Updated last month
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,373Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,563Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,347Jul 3, 2024Updated last year
- PyTorch extensions for high performance and large scale training.☆3,403Apr 26, 2025Updated 10 months ago
- Hummingbird compiles trained ML models into tensor computation for faster inference.☆3,531Jul 17, 2025Updated 8 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,504Mar 13, 2026Updated last week
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,033Mar 14, 2026Updated last week
- Fast and memory-efficient exact attention☆22,832Updated this week
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,266Mar 27, 2024Updated last year
- A collection of libraries to optimise AI model performances☆8,350Jul 22, 2024Updated last year
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,730Feb 26, 2026Updated 3 weeks ago
- Prune a model while finetuning or training.☆406Jun 21, 2022Updated 3 years ago
- [MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration☆3,463Jul 17, 2025Updated 8 months ago