Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
☆2,144Jun 2, 2025Updated 8 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,159Jun 2, 2025Updated 8 months ago
- Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes☆387Jun 2, 2025Updated 8 months ago
- ML model optimization product to accelerate inference.☆326Jun 2, 2025Updated 8 months ago
- Top-level directory for documentation and general content☆120Jun 2, 2025Updated 8 months ago
- SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, …☆2,590Updated this week
- A data augmentations library for audio, image, text, and video.☆5,071Feb 13, 2026Updated 2 weeks ago
- Accessible large language models via k-bit quantization for PyTorch.☆7,997Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Feb 21, 2026Updated last week
- 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization…☆3,296Feb 9, 2026Updated 2 weeks ago
- Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stre…☆9,012Feb 16, 2026Updated last week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,986Jun 16, 2024Updated last year
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆870Aug 20, 2024Updated last year
- Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains☆1,733Oct 8, 2023Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀☆1,687Oct 23, 2024Updated last year
- PyTorch native quantization and sparsity for training and inference☆2,696Feb 22, 2026Updated last week
- 🐍 Geometric Computer Vision Library for Spatial AI☆11,093Updated this week
- Development repository for the Triton language and compiler☆18,460Feb 22, 2026Updated last week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,033Jan 23, 2026Updated last month
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week
- Hummingbird compiles trained ML models into tensor computation for faster inference.☆3,529Jul 17, 2025Updated 7 months ago
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆14,339Jul 3, 2024Updated last year
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,006Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,353Feb 20, 2026Updated last week
- AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.☆2,563Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,513Updated this week
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- A collection of libraries to optimise AI model performances☆8,353Jul 22, 2024Updated last year
- [CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.☆3,258Sep 7, 2025Updated 5 months ago
- Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data …☆11,333Jan 13, 2026Updated last month
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,724Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,397Updated this week
- The easiest way to serve AI apps and models - Build Model Inference APIs, Job queues, LLM apps, Multi-model pipelines, and more!☆8,472Updated this week
- ⚡ Build your chatbot within minutes on your favorite device; offer SOTA compression techniques for LLMs; run LLMs efficiently on Intel Pl…☆2,175Oct 8, 2024Updated last year
- Fast and memory-efficient exact attention☆22,361Updated this week
- Lightweight Python library for adding real-time multi-object tracking to any detector.☆2,618Apr 30, 2025Updated 10 months ago
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,585Jan 28, 2026Updated last month
- Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.☆5,007Updated this week