All modules for which code is available
- torchao.float8.config
- torchao.float8.float8_linear_utils
- torchao.float8.fsdp_utils
- torchao.prototype.mx_formats.inference_workflow
- torchao.prototype.quantization.quant_api
- torchao.quantization.qat.api
- torchao.quantization.qat.embedding
- torchao.quantization.qat.fake_quantize_config
- torchao.quantization.qat.fake_quantizer
- torchao.quantization.qat.linear
- torchao.quantization.quant_api
- torchao.quantization.quantize_.common.kernel_preference
- torchao.quantization.quantize_.common.packing_format
- torchao.quantization.quantize_.common.quantize_tensor_kwargs
- torchao.sparsity.sparse_api
- torchao.sparsity.utils
- torchao.sparsity.wanda
- torchao.utils