About Anaconda Help Download Anaconda

PyTorch native quantization and sparsity for training and inference

copied from cf-post-staging / torchao

Installers

  • linux-64 v0.12.0
  • osx-64 v0.12.0

conda install

To install this package run one of the following:
conda install conda-forge::torchao

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.


© 2025 Anaconda, Inc. All Rights Reserved. (v4.2.1) Legal | Privacy Policy