JAX quantization library used for PokeBNN project - View it on GitHub