.pre-commit-config.yaml
LICENSE
MANIFEST.in
README.md
setup.cfg
setup.py
notebooks/01_quant_tensor_quant_conv2d_overview.ipynb
notebooks/02_quant_activation_overview.ipynb
notebooks/03_anatomy_of_a_quantizer.ipynb
notebooks/Brevitas_TVMCon2021.ipynb
notebooks/ONNX_export_tutorial.ipynb
notebooks/minifloat_mx_tutorial.ipynb
notebooks/quantized_recurrent.ipynb
requirements/requirements-dev.txt
requirements/requirements-docs.txt
requirements/requirements-export.txt
requirements/requirements-finn-integration.txt
requirements/requirements-hadamard.txt
requirements/requirements-llm.txt
requirements/requirements-notebook.txt
requirements/requirements-nox.txt
requirements/requirements-ort-integration.txt
requirements/requirements-setup.txt
requirements/requirements-stt.txt
requirements/requirements-test.txt
requirements/requirements-tts.txt
requirements/requirements-vision.txt
requirements/requirements.txt
src/brevitas/__init__.py
src/brevitas/common.py
src/brevitas/config.py
src/brevitas/jit.py
src/brevitas/test_qt.py
src/brevitas.egg-info/PKG-INFO
src/brevitas.egg-info/SOURCES.txt
src/brevitas.egg-info/dependency_links.txt
src/brevitas.egg-info/entry_points.txt
src/brevitas.egg-info/not-zip-safe
src/brevitas.egg-info/requires.txt
src/brevitas.egg-info/top_level.txt
src/brevitas/backport/__init__.py
src/brevitas/backport/_guards.py
src/brevitas/backport/_ops.py
src/brevitas/backport/fx/__init__.py
src/brevitas/backport/fx/_compatibility.py
src/brevitas/backport/fx/_pytree.py
src/brevitas/backport/fx/_symbolic_trace.py
src/brevitas/backport/fx/annotate.py
src/brevitas/backport/fx/config.py
src/brevitas/backport/fx/graph.py
src/brevitas/backport/fx/graph_module.py
src/brevitas/backport/fx/immutable_collections.py
src/brevitas/backport/fx/interpreter.py
src/brevitas/backport/fx/node.py
src/brevitas/backport/fx/operator_schemas.py
src/brevitas/backport/fx/proxy.py
src/brevitas/backport/fx/subgraph_rewriter.py
src/brevitas/backport/fx/traceback.py
src/brevitas/backport/fx/experimental/proxy_tensor.py
src/brevitas/backport/fx/experimental/symbolic_shapes.py
src/brevitas/backport/fx/passes/__init__.py
src/brevitas/backport/fx/passes/shape_prop.py
src/brevitas/backport/utils/_stats.py
src/brevitas/backport/utils/weak.py
src/brevitas/core/__init__.py
src/brevitas/core/restrict_val.py
src/brevitas/core/utils.py
src/brevitas/core/zero_point.py
src/brevitas/core/bit_width/__init__.py
src/brevitas/core/bit_width/const.py
src/brevitas/core/bit_width/parameter.py
src/brevitas/core/function_wrapper/__init__.py
src/brevitas/core/function_wrapper/clamp.py
src/brevitas/core/function_wrapper/learned_round.py
src/brevitas/core/function_wrapper/misc.py
src/brevitas/core/function_wrapper/ops_ste.py
src/brevitas/core/function_wrapper/shape.py
src/brevitas/core/function_wrapper/stochastic_round.py
src/brevitas/core/quant/__init__.py
src/brevitas/core/quant/binary.py
src/brevitas/core/quant/delay.py
src/brevitas/core/quant/float.py
src/brevitas/core/quant/int.py
src/brevitas/core/quant/int_base.py
src/brevitas/core/quant/ternary.py
src/brevitas/core/scaling/__init__.py
src/brevitas/core/scaling/float_scaling.py
src/brevitas/core/scaling/int_scaling.py
src/brevitas/core/scaling/pre_scaling.py
src/brevitas/core/scaling/runtime.py
src/brevitas/core/scaling/standalone.py
src/brevitas/core/scaling/test.py
src/brevitas/core/stats/__init__.py
src/brevitas/core/stats/stats_op.py
src/brevitas/core/stats/stats_wrapper.py
src/brevitas/core/stats/view_wrapper.py
src/brevitas/csrc/autograd_ste_ops.cpp
src/brevitas/export/__init__.py
src/brevitas/export/manager.py
src/brevitas/export/common/__init__.py
src/brevitas/export/common/handler/__init__.py
src/brevitas/export/common/handler/base.py
src/brevitas/export/common/handler/qcdq.py
src/brevitas/export/inference/__init__.py
src/brevitas/export/inference/handler.py
src/brevitas/export/inference/manager.py
src/brevitas/export/onnx/__init__.py
src/brevitas/export/onnx/debug.py
src/brevitas/export/onnx/function.py
src/brevitas/export/onnx/handler.py
src/brevitas/export/onnx/manager.py
src/brevitas/export/onnx/qonnx/__init__.py
src/brevitas/export/onnx/qonnx/function.py
src/brevitas/export/onnx/qonnx/handler.py
src/brevitas/export/onnx/qonnx/manager.py
src/brevitas/export/onnx/standard/__init__.py
src/brevitas/export/onnx/standard/function.py
src/brevitas/export/onnx/standard/manager.py
src/brevitas/export/onnx/standard/qcdq/__init__.py
src/brevitas/export/onnx/standard/qcdq/handler.py
src/brevitas/export/onnx/standard/qcdq/manager.py
src/brevitas/export/torch/qcdq/__init__.py
src/brevitas/export/torch/qcdq/handler.py
src/brevitas/export/torch/qcdq/manager.py
src/brevitas/function/__init__.py
src/brevitas/function/ops.py
src/brevitas/function/ops_ste.py
src/brevitas/function/shape.py
src/brevitas/fx/__init__.py
src/brevitas/fx/brevitas_tracer.py
src/brevitas/fx/value_tracer.py
src/brevitas/graph/__init__.py
src/brevitas/graph/base.py
src/brevitas/graph/calibrate.py
src/brevitas/graph/channel_splitting.py
src/brevitas/graph/equalize.py
src/brevitas/graph/fixed_point.py
src/brevitas/graph/gpfq.py
src/brevitas/graph/gptq.py
src/brevitas/graph/gpxq.py
src/brevitas/graph/per_input.py
src/brevitas/graph/quantize.py
src/brevitas/graph/quantize_impl.py
src/brevitas/graph/standardize.py
src/brevitas/graph/utils.py
src/brevitas/graph/target/__init__.py
src/brevitas/graph/target/flexml.py
src/brevitas/inject/__init__.py
src/brevitas/inject/defaults.py
src/brevitas/inject/enum.py
src/brevitas/loss/__init__.py
src/brevitas/loss/base_loss.py
src/brevitas/loss/weighted_bit_width.py
src/brevitas/nn/__init__.py
src/brevitas/nn/equalized_layer.py
src/brevitas/nn/hadamard_classifier.py
src/brevitas/nn/quant_accumulator.py
src/brevitas/nn/quant_activation.py
src/brevitas/nn/quant_avg_pool.py
src/brevitas/nn/quant_bn.py
src/brevitas/nn/quant_conv.py
src/brevitas/nn/quant_convtranspose.py
src/brevitas/nn/quant_eltwise.py
src/brevitas/nn/quant_embedding.py
src/brevitas/nn/quant_layer.py
src/brevitas/nn/quant_linear.py
src/brevitas/nn/quant_mha.py
src/brevitas/nn/quant_rnn.py
src/brevitas/nn/quant_scale_bias.py
src/brevitas/nn/quant_upsample.py
src/brevitas/nn/utils.py
src/brevitas/nn/mixin/__init__.py
src/brevitas/nn/mixin/acc.py
src/brevitas/nn/mixin/act.py
src/brevitas/nn/mixin/base.py
src/brevitas/nn/mixin/parameter.py
src/brevitas/nn/mixin/utils.py
src/brevitas/nn/target/__init__.py
src/brevitas/nn/target/flexml.py
src/brevitas/onnx/__init__.py
src/brevitas/ops/__init__.py
src/brevitas/ops/autograd_ste_ops.py
src/brevitas/proxy/__init__.py
src/brevitas/proxy/float_parameter_quant.py
src/brevitas/proxy/float_runtime_quant.py
src/brevitas/proxy/groupwise_float_parameter_quant.py
src/brevitas/proxy/groupwise_float_runtime_quant.py
src/brevitas/proxy/groupwise_int_parameter_quant.py
src/brevitas/proxy/groupwise_int_runtime_quant.py
src/brevitas/proxy/parameter_quant.py
src/brevitas/proxy/quant_proxy.py
src/brevitas/proxy/runtime_quant.py
src/brevitas/proxy/utils.py
src/brevitas/quant/__init__.py
src/brevitas/quant/base.py
src/brevitas/quant/binary.py
src/brevitas/quant/fixed_point.py
src/brevitas/quant/none.py
src/brevitas/quant/scaled_int.py
src/brevitas/quant/shifted_scaled_int.py
src/brevitas/quant/ternary.py
src/brevitas/quant/experimental/__init__.py
src/brevitas/quant/experimental/float.py
src/brevitas/quant/experimental/float_base.py
src/brevitas/quant/experimental/float_quant_fnuz.py
src/brevitas/quant/experimental/float_quant_ocp.py
src/brevitas/quant/experimental/mx_quant_ocp.py
src/brevitas/quant/solver/__init__.py
src/brevitas/quant/solver/act.py
src/brevitas/quant/solver/bias.py
src/brevitas/quant/solver/clamp.py
src/brevitas/quant/solver/common.py
src/brevitas/quant/solver/parameter.py
src/brevitas/quant/solver/trunc.py
src/brevitas/quant/solver/weight.py
src/brevitas/quant_tensor/__init__.py
src/brevitas/quant_tensor/base_quant_tensor.py
src/brevitas/quant_tensor/float_quant_tensor.py
src/brevitas/quant_tensor/float_torch_handler.py
src/brevitas/quant_tensor/groupwise_float_quant_tensor.py
src/brevitas/quant_tensor/groupwise_int_quant_tensor.py
src/brevitas/quant_tensor/int_quant_tensor.py
src/brevitas/quant_tensor/int_torch_handler.py
src/brevitas/quant_tensor/test.py
src/brevitas/quant_tensor/torch_handler.py
src/brevitas/utils/__init__.py
src/brevitas/utils/float_quant_utils.py
src/brevitas/utils/jit_utils.py
src/brevitas/utils/logging.py
src/brevitas/utils/python_utils.py
src/brevitas/utils/quant_utils.py
src/brevitas/utils/torch_utils.py
src/brevitas_examples/README.md
src/brevitas_examples/__init__.py
src/brevitas_examples/float_example.py
src/brevitas_examples/bnn_pynq/README.md
src/brevitas_examples/bnn_pynq/__init__.py
src/brevitas_examples/bnn_pynq/bnn_pynq_train.py
src/brevitas_examples/bnn_pynq/logger.py
src/brevitas_examples/bnn_pynq/trainer.py
src/brevitas_examples/bnn_pynq/cfg/__init__.py
src/brevitas_examples/bnn_pynq/cfg/cnv_1w1a.ini
src/brevitas_examples/bnn_pynq/cfg/cnv_1w2a.ini
src/brevitas_examples/bnn_pynq/cfg/cnv_2w2a.ini
src/brevitas_examples/bnn_pynq/cfg/lfc_1w1a.ini
src/brevitas_examples/bnn_pynq/cfg/lfc_1w2a.ini
src/brevitas_examples/bnn_pynq/cfg/resnet18_4w4a.ini
src/brevitas_examples/bnn_pynq/cfg/sfc_1w1a.ini
src/brevitas_examples/bnn_pynq/cfg/sfc_1w2a.ini
src/brevitas_examples/bnn_pynq/cfg/sfc_2w2a.ini
src/brevitas_examples/bnn_pynq/cfg/tfc_1w1a.ini
src/brevitas_examples/bnn_pynq/cfg/tfc_1w2a.ini
src/brevitas_examples/bnn_pynq/cfg/tfc_2w2a.ini
src/brevitas_examples/bnn_pynq/models/CNV.py
src/brevitas_examples/bnn_pynq/models/FC.py
src/brevitas_examples/bnn_pynq/models/__init__.py
src/brevitas_examples/bnn_pynq/models/common.py
src/brevitas_examples/bnn_pynq/models/losses.py
src/brevitas_examples/bnn_pynq/models/resnet.py
src/brevitas_examples/bnn_pynq/models/tensor_norm.py
src/brevitas_examples/common/parse_utils.py
src/brevitas_examples/common/generative/nn.py
src/brevitas_examples/common/generative/quant_blocks.py
src/brevitas_examples/common/generative/quantize.py
src/brevitas_examples/common/generative/quantizers.py
src/brevitas_examples/imagenet_classification/README.md
src/brevitas_examples/imagenet_classification/__init__.py
src/brevitas_examples/imagenet_classification/utils.py
src/brevitas_examples/imagenet_classification/a2q/README.md
src/brevitas_examples/imagenet_classification/a2q/__init__.py
src/brevitas_examples/imagenet_classification/a2q/a2q_evaluate_models.py
src/brevitas_examples/imagenet_classification/a2q/a2q_train_models.py
src/brevitas_examples/imagenet_classification/a2q/ep_init.py
src/brevitas_examples/imagenet_classification/a2q/quant.py
src/brevitas_examples/imagenet_classification/a2q/resnet.py
src/brevitas_examples/imagenet_classification/a2q/utils.py
src/brevitas_examples/imagenet_classification/models/README.md
src/brevitas_examples/imagenet_classification/models/__init__.py
src/brevitas_examples/imagenet_classification/models/common.py
src/brevitas_examples/imagenet_classification/models/mobilenetv1.py
src/brevitas_examples/imagenet_classification/models/proxylessnas.py
src/brevitas_examples/imagenet_classification/models/vgg.py
src/brevitas_examples/imagenet_classification/models/cfg/__init__.py
src/brevitas_examples/imagenet_classification/models/cfg/quant_mobilenet_v1.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_mobilenet_v1_4b.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_mobilenet_v1_4b_round_avgpool.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_proxylessnas_mobile14_4b.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_proxylessnas_mobile14_4b5b.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_proxylessnas_mobile14_4b_round_avgpool.ini
src/brevitas_examples/imagenet_classification/models/cfg/quant_proxylessnas_mobile14_hadamard_4b.ini
src/brevitas_examples/imagenet_classification/ptq/README.md
src/brevitas_examples/imagenet_classification/ptq/RESULTS_IMGCLSMOB.csv
src/brevitas_examples/imagenet_classification/ptq/RESULTS_TORCHVISION.csv
src/brevitas_examples/imagenet_classification/ptq/RESULTS_TORCHVISION_BEST_CONFIGS.csv
src/brevitas_examples/imagenet_classification/ptq/learned_round_utils.py
src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
src/brevitas_examples/imagenet_classification/ptq/utils.py
src/brevitas_examples/imagenet_classification/ptq/benchmark/parallel.sh
src/brevitas_examples/imagenet_classification/ptq/benchmark/post_processing.py
src/brevitas_examples/imagenet_classification/ptq/benchmark/ptq_benchmark_quant_model.py
src/brevitas_examples/imagenet_classification/ptq/benchmark/ptq_benchmark_torchvision.py
src/brevitas_examples/imagenet_classification/ptq/benchmark/single_command.sh
src/brevitas_examples/imagenet_classification/qat/README.md
src/brevitas_examples/imagenet_classification/qat/__init__.py
src/brevitas_examples/imagenet_classification/qat/imagenet_val.py
src/brevitas_examples/llm/README.md
src/brevitas_examples/llm/__init__.py
src/brevitas_examples/llm/main.py
src/brevitas_examples/llm/pergroup_qdq.py
src/brevitas_examples/llm/test.py
src/brevitas_examples/llm/test_linear_mlir_export.py
src/brevitas_examples/llm/llm_quant/__init__.py
src/brevitas_examples/llm/llm_quant/bias_corr.py
src/brevitas_examples/llm/llm_quant/calibrate.py
src/brevitas_examples/llm/llm_quant/data.py
src/brevitas_examples/llm/llm_quant/data_utils.py
src/brevitas_examples/llm/llm_quant/equalize.py
src/brevitas_examples/llm/llm_quant/eval.py
src/brevitas_examples/llm/llm_quant/export.py
src/brevitas_examples/llm/llm_quant/gpxq.py
src/brevitas_examples/llm/llm_quant/ln_affine_merge.py
src/brevitas_examples/llm/llm_quant/mha_layers.py
src/brevitas_examples/llm/llm_quant/mlir_custom_mm.py
src/brevitas_examples/llm/llm_quant/prepare_for_quantize.py
src/brevitas_examples/llm/llm_quant/quantize.py
src/brevitas_examples/llm/llm_quant/run_utils.py
src/brevitas_examples/llm/llm_quant/sharded_mlir_group_export.py
src/brevitas_examples/llm/llm_quant/test.py
src/brevitas_examples/llm/llm_quant/awq/auto_clip.py
src/brevitas_examples/llm/llm_quant/awq/auto_scale.py
src/brevitas_examples/llm/llm_quant/awq/pre_quant.py
src/brevitas_examples/llm/llm_quant/awq/quantizer.py
src/brevitas_examples/llm/llm_quant/awq/utils/calib_data.py
src/brevitas_examples/llm/llm_quant/awq/utils/module.py
src/brevitas_examples/speech_to_text/README.md
src/brevitas_examples/speech_to_text/__init__.py
src/brevitas_examples/speech_to_text/get_librispeech_data.py
src/brevitas_examples/speech_to_text/quartznet_val.py
src/brevitas_examples/speech_to_text/cfg/__init__.py
src/brevitas_examples/speech_to_text/cfg/quant_quartznet_perchannelscaling_4b.ini
src/brevitas_examples/speech_to_text/cfg/quant_quartznet_perchannelscaling_8b.ini
src/brevitas_examples/speech_to_text/cfg/quant_quartznet_pertensorscaling_8b.ini
src/brevitas_examples/speech_to_text/cfg/topology/__init__.py
src/brevitas_examples/speech_to_text/cfg/topology/quartznet15x5.yaml
src/brevitas_examples/speech_to_text/quartznet/__init__.py
src/brevitas_examples/speech_to_text/quartznet/audio_preprocessing.py
src/brevitas_examples/speech_to_text/quartznet/data_layer.py
src/brevitas_examples/speech_to_text/quartznet/greedy_ctc_decoder.py
src/brevitas_examples/speech_to_text/quartznet/helpers.py
src/brevitas_examples/speech_to_text/quartznet/losses.py
src/brevitas_examples/speech_to_text/quartznet/metrics.py
src/brevitas_examples/speech_to_text/quartznet/quartznet.py
src/brevitas_examples/speech_to_text/quartznet/parts/__init__.py
src/brevitas_examples/speech_to_text/quartznet/parts/cleaners.py
src/brevitas_examples/speech_to_text/quartznet/parts/common.py
src/brevitas_examples/speech_to_text/quartznet/parts/dataset.py
src/brevitas_examples/speech_to_text/quartznet/parts/features.py
src/brevitas_examples/speech_to_text/quartznet/parts/manifest.py
src/brevitas_examples/speech_to_text/quartznet/parts/perturb.py
src/brevitas_examples/speech_to_text/quartznet/parts/quartznet.py
src/brevitas_examples/speech_to_text/quartznet/parts/segment.py
src/brevitas_examples/speech_to_text/quartznet/parts/spectr_augment.py
src/brevitas_examples/stable_diffusion/README.md
src/brevitas_examples/stable_diffusion/__init__.py
src/brevitas_examples/stable_diffusion/get_outputs.py
src/brevitas_examples/stable_diffusion/main.py
src/brevitas_examples/stable_diffusion/test_compile.py
src/brevitas_examples/stable_diffusion/mlperf_evaluation/accuracy.py
src/brevitas_examples/stable_diffusion/mlperf_evaluation/backend.py
src/brevitas_examples/stable_diffusion/mlperf_evaluation/dataset.py
src/brevitas_examples/stable_diffusion/mlperf_evaluation/inception.py
src/brevitas_examples/stable_diffusion/mlperf_evaluation/requirements.txt
src/brevitas_examples/stable_diffusion/sd_quant/__init__.py
src/brevitas_examples/stable_diffusion/sd_quant/constants.py
src/brevitas_examples/stable_diffusion/sd_quant/export.py
src/brevitas_examples/stable_diffusion/sd_quant/nn.py
src/brevitas_examples/stable_diffusion/sd_quant/utils.py
src/brevitas_examples/super_resolution/README.md
src/brevitas_examples/super_resolution/eval_model.py
src/brevitas_examples/super_resolution/train_model.py
src/brevitas_examples/super_resolution/models/__init__.py
src/brevitas_examples/super_resolution/models/common.py
src/brevitas_examples/super_resolution/models/espcn.py
src/brevitas_examples/super_resolution/utils/__init__.py
src/brevitas_examples/super_resolution/utils/dataset.py
src/brevitas_examples/super_resolution/utils/evaluate.py
src/brevitas_examples/super_resolution/utils/export.py
src/brevitas_examples/super_resolution/utils/train.py
src/brevitas_examples/text_to_speech/README.md
src/brevitas_examples/text_to_speech/__init__.py
src/brevitas_examples/text_to_speech/melgan_val.py
src/brevitas_examples/text_to_speech/preprocess_dataset.py
src/brevitas_examples/text_to_speech/cfg/__init__.py
src/brevitas_examples/text_to_speech/cfg/quant_melgan_8b.ini
src/brevitas_examples/text_to_speech/melgan/__init__.py
src/brevitas_examples/text_to_speech/melgan/common.py
src/brevitas_examples/text_to_speech/melgan/generator_brevitas.py
src/brevitas_examples/text_to_speech/melgan/melgan.py
src/brevitas_examples/text_to_speech/melgan/res_stack_brevitas.py
src/brevitas_examples/text_to_speech/utilities/__init__.py
src/brevitas_examples/text_to_speech/utilities/audio_processing.py
src/brevitas_examples/text_to_speech/utilities/stft.py