search

approx-ml / approx

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

Replace `numpy` with `JAX` #23

Open ma7dev opened 2 years ago

ma7dev commented 2 years ago

Describe the solution you'd like We can't use numpy to be quantized because everyone is using frameworks to define their models, so it would make sense to use JAX instead as a 2nd backend to be supported.