欢迎来到 torchao 文档

torchao 是一个用于自定义数据类型和优化的库。使用原生 PyTorch 量化和稀疏化权重、梯度、优化器和激活，用于推理和训练。请查看 torchao README，以获得对库的总体介绍以及最近的亮点和更新。此处的文档将重点介绍

1. API 参考

2. 开发者贡献指南

3. 教程

API 参考

• torchao.dtypes
• torchao.quantization
• torchao.sparsity

贡献者指南

• torchao 贡献者指南

教程

• 序列化