The torchao package contains apis and workflows used to apply AO techniques like quantization and pruning to GPU models using only native pytorch.
tbd
tbd
torchao is released under the BSD 3 license.
forked from pytorch/ao
-
Notifications
You must be signed in to change notification settings - Fork 0
PyTorch native quantization and sparsity for training and inference
License
upskyy/ao
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
PyTorch native quantization and sparsity for training and inference
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 83.5%
- C++ 10.0%
- Cuda 4.3%
- Shell 0.8%
- Metal 0.5%
- CMake 0.5%
- Other 0.4%