approx-ml / approx

Automatic quantization library
https://approx-ml.github.io/approx/
Apache License 2.0
11 stars 1 forks source link

Implement `approx.auto_quantize()` function #16

Closed bushshrub closed 2 years ago

bushshrub commented 2 years ago

Describe the solution you'd like

Adds support for a public API function, approx.auto_quantize(model, pretrained: bool)

This accepts some model and automatically quantizes it

Additional context

This replaces #3 temporarily for a nicer user-facing API

Additional improvements could include supporting an @auto_quantize annotation over some given model