neuralmagic / sparseml
Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
☆2,071Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for sparseml
- Sparsity-aware deep learning inference runtime for CPUs☆3,028Updated 4 months ago
- Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes☆370Updated 4 months ago
- Top-level directory for documentation and general content☆120Updated 4 months ago
- ML model optimization product to accelerate inference.☆320Updated 7 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,535Updated 9 months ago
- Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀☆1,659Updated 3 weeks ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,471Updated 2 months ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,134Updated this week
- SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX R…☆2,227Updated this 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,743Updated last year
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,867Updated 5 months ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,011Updated 7 months ago
- Library for 8-bit optimizers and quantization routines.☆714Updated 2 years ago
- 🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools☆2,576Updated this week
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆2,328Updated last month
- PyTorch extensions for high performance and large scale training.☆3,195Updated last week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,597Updated this week
- AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.☆2,148Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,396Updated this week
- Actively maintained ONNX Optimizer☆647Updated 8 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆1,979Updated this week
- Hummingbird compiles trained ML models into tensor computation for faster inference.☆3,356Updated this week
- Neural Network Compression Framework for enhanced OpenVINO™ inference☆943Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,408Updated 3 months ago
- Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.☆715Updated 3 weeks ago
- An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come☆849Updated this week
- 10x faster matrix and vector operations☆2,474Updated 2 years ago
- maximal update parametrization (µP)☆1,402Updated 4 months ago
- IDE for PyTorch and its ecosystem☆383Updated 6 months ago