Add files via upload

diffq/__init__.py

@@ -0,0 +1,29 @@
+# Copyright (c) Facebook, Inc. and its affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+
+# flake8: noqa
+"""
+This package implements different quantization strategies:
+
+- `diffq.uniform.UniformQuantizer`: classic uniform quantization over n bits.
+- `diffq.diffq.DiffQuantizer`: differentiable quantizer based on scaled noise injection.
+- `diffq.lsq.LSQ`: Learnt Step size Quantizer based on [Esser et al. 2019] https://arxiv.org/abs/1902.08153
+- `diffq.bitpack`: efficient CPU bit-packing for returning quantized states.
+- `diffq.torch_pack`: torch based bit-packing compatible with torchscript.
+- `diffq.ts_export`: support exporting DiffQ based models to torchscript.
+
+
+Also, do check `diffq.base.BaseQuantizer` for the common methods of all Quantizers.
+"""
+
+from .uniform import UniformQuantizer
+from .diffq import DiffQuantizer
+from .lsq import LSQ
+from .base import restore_quantized_state
+from . import ts_export
+
+
+__version__ = "0.2.2"