CMD + K

torchao

Community

PyTorch native quantization and sparsity for training and inference

Installation

To install this package, run one of the following:

Conda
$conda install conda-forge::torchao

Usage Tracking

0.13.0
0.12.0
2 / 8 versions selected
Downloads (Last 6 months): 0

Description

TorchAO is a PyTorch-native model optimization framework leveraging quantization and sparsity to provide an end-to-end, training-to-serving workflow for AI models. TorchAO works out-of-the-box with torch.compile() and FSDP2 across most HuggingFace PyTorch models.

About

Summary

PyTorch native quantization and sparsity for training and inference

Last Updated

Oct 20, 2025 at 19:52

License

BSD-3-Clause

Supported Platforms

macOS-64
linux-64