Metadata-Version: 2.4
Name: fp_quant
Version: 0.3.2
Summary: A Python library for FP-Quant
Home-page: https://github.com/IST-DASLab/FP-Quant
Author: Andrei Panferov
Author-email: andrei@panferov.org
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.9
Description-Content-Type: text/markdown
Requires-Dist: torch>=2.8.0
Requires-Dist: scipy>=1.13.0
Requires-Dist: triton>=3.3.0
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

# fp_quant

A library that wraps [`qutlass`](https://github.com/IST-DASLab/qutlass) kernels with linear layer wrappers for integrations into training and inference engines.

## Installation

```bash
pip install .
```

## Usage

```python
from fp_quant import replace_with_fp_quant_linear, FPQuantConfig

# Replace nn.Linear layers with fp_quant.FPQuantLinear
replace_with_fp_quant_linear(
    model,
    fp_quant_linear_config=FPQuantConfig(),
)
``` 
