Installation

TorchQuant can be installed using pip, the Python package installer. To install TorchQuant, run the following command:

pip install --upgrade torchquantlib

Requirements

TorchQuant requires:

• Python 3.7+

• PyTorch 2.4.0+

• NumPy

• Geomloss

For GPU acceleration, ensure you have a CUDA-compatible GPU and the appropriate CUDA toolkit installed.