neuralmagic / sparseml
Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
☆2,089Updated 5 months ago
Alternatives and similar repositories for sparseml:
Users that are interested in sparseml are comparing it to the libraries listed below
- Sparsity-aware deep learning inference runtime for CPUs☆3,077Updated 5 months ago
- Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes☆376Updated 5 months ago
- Top-level directory for documentation and general content☆120Updated last month
- ML model optimization product to accelerate inference.☆322Updated 9 months ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,880Updated 7 months ago
- D2Go is a toolkit for efficient deep learning☆844Updated 2 months ago
- PyTorch extensions for high performance and large scale training.☆3,232Updated this week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,016Updated 9 months ago
- Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀☆1,671Updated 2 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,549Updated 11 months ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,426Updated last month
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,823Updated 3 weeks ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,647Updated this week
- MADGRAD Optimization Method☆803Updated 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,746Updated last year
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,496Updated 4 months ago
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆2,516Updated 3 weeks ago
- Hummingbird compiles trained ML models into tensor computation for faster inference.☆3,372Updated last week
- Code release for "Detecting Twenty-thousand Classes using Image-level Supervision".☆1,905Updated 9 months ago
- SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX R…☆2,298Updated this week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,160Updated this week
- Cramming the training of a (BERT-type) language model into limited compute.☆1,307Updated 7 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,403Updated this week
- 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization…☆2,667Updated this week
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,704Updated last week
- A deep learning library for video understanding research.☆3,365Updated last month
- Foundation Architecture for (M)LLMs☆3,038Updated 9 months ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,528Updated this week