torchao
Subscription: Developer
Active
Subscription: Developer
A PyTorch library for applying architecture optimization techniques to GPU models, including quantization, sparsity, and low-rank approximations to reduce memory usage and improve inference and training performance of deep learning workloads.
pip install torchao==0.18.0
0.18.0
Linux