.pre-commit-config.yaml
CONTRIBUTING.md
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-diffusion.txt
requirements/requirements-docs.txt
requirements/requirements-export.txt
requirements/requirements-finn-integration.txt
requirements/requirements-hadamard.txt
requirements/requirements-lighteval.txt
requirements/requirements-llm.txt
requirements/requirements-notebook.txt
requirements/requirements-nox.txt
requirements/requirements-numpy.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/compiler.py
src/brevitas/config.py
src/brevitas/jit.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/scm_file_list.json
src/brevitas.egg-info/scm_version.json
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/float.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/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/inference/vLLM/__init__.py
src/brevitas/export/inference/vLLM/handler.py
src/brevitas/export/inference/vLLM/layer.py
src/brevitas/export/inference/vLLM/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/shark/handler.py
src/brevitas/export/shark/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/hadamard.py
src/brevitas/graph/hadamard_tensors.pt
src/brevitas/graph/magr.py
src/brevitas/graph/per_input.py
src/brevitas/graph/permute.py
src/brevitas/graph/qronos.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/library/__init__.py
src/brevitas/library/custom_op.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_sdpa.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/optim/__init__.py
src/brevitas/optim/cailey_sgd.py
src/brevitas/optim/sign_sgd.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/float.py
src/brevitas/quant/float_base.py
src/brevitas/quant/float_quant_fnuz.py
src/brevitas/quant/float_quant_ocp.py
src/brevitas/quant/mx_quant_ocp.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/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/parametrization_utils.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/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/axe.py
src/brevitas_examples/common/dynamo_utils.py
src/brevitas_examples/common/parse_utils.py
src/brevitas_examples/common/svd_quant.py
src/brevitas_examples/common/trainer_utils.py
src/brevitas_examples/common/accelerate_utils/accelerate.py
src/brevitas_examples/common/benchmark/cfg2benchmarkcfg.py
src/brevitas_examples/common/benchmark/cfg2randbenchmarkcfg.py
src/brevitas_examples/common/benchmark/gen_template.py
src/brevitas_examples/common/benchmark/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/common/learned_round/learned_round_args.py
src/brevitas_examples/common/learned_round/learned_round_method.py
src/brevitas_examples/common/learned_round/learned_round_trainer.py
src/brevitas_examples/common/learned_round/learned_round_utils.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/ptq_imagenet_args.py
src/brevitas_examples/imagenet_classification/ptq/utils.py
src/brevitas_examples/imagenet_classification/ptq/benchmark/benchmark_template.yaml
src/brevitas_examples/imagenet_classification/ptq/benchmark/ptq_benchmark.py
src/brevitas_examples/imagenet_classification/ptq/config/default_template.yaml
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/eval_lighteval.py
src/brevitas_examples/llm/llm_args.py
src/brevitas_examples/llm/main.py
src/brevitas_examples/llm/benchmark/benchmark_rand_template.yaml
src/brevitas_examples/llm/benchmark/benchmark_template.yaml
src/brevitas_examples/llm/benchmark/llm_benchmark.py
src/brevitas_examples/llm/benchmark/llm_rand_benchmark.py
src/brevitas_examples/llm/config/default_template.yaml
src/brevitas_examples/llm/config/learned_round_template.yaml
src/brevitas_examples/llm/gguf_export/convert.py
src/brevitas_examples/llm/gguf_export/export.py
src/brevitas_examples/llm/gguf_export/quant.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/learned_round_utils.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/parse_utils.py
src/brevitas_examples/llm/llm_quant/prepare_for_quantize.py
src/brevitas_examples/llm/llm_quant/rotation_optimization.py
src/brevitas_examples/llm/llm_quant/run_utils.py
src/brevitas_examples/llm/llm_quant/svd_quant.py
src/brevitas_examples/llm/llm_quant/trainer_utils.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/graph.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/llm/llm_quant/awq/utils/region.py
src/brevitas_examples/papers/axe/README.md
src/brevitas_examples/papers/axe/benchmark-llama3-channelwise.yml
src/brevitas_examples/papers/axe/benchmark-llama3-groupwise.yml
src/brevitas_examples/papers/axe/benchmark.py
src/brevitas_examples/papers/expansion/README.md
src/brevitas_examples/papers/expansion/benchmark.py
src/brevitas_examples/papers/expansion/double_quarot_star.yaml
src/brevitas_examples/papers/expansion/double_spinquant_star.yaml
src/brevitas_examples/papers/expansion/quarot_star.yaml
src/brevitas_examples/papers/expansion/quarot_star_mxfp4.yaml
src/brevitas_examples/papers/expansion/spinquant_star.yaml
src/brevitas_examples/papers/expansion/spinquant_star_mxfp4.yaml
src/brevitas_examples/papers/learned_round/README.md
src/brevitas_examples/papers/learned_round/benchmark.py
src/brevitas_examples/papers/learned_round/learned_round_mxfp4.yaml
src/brevitas_examples/papers/learned_round/learned_round_mxfp4_spinquant.yaml
src/brevitas_examples/papers/learned_round/learned_round_weight_act.yaml
src/brevitas_examples/papers/learned_round/learned_round_weight_act_spinquant.yaml
src/brevitas_examples/papers/learned_round/learned_round_weight_only.yaml
src/brevitas_examples/papers/learned_round/learned_round_weight_only_spinquant.yaml
src/brevitas_examples/papers/perq/README.md
src/brevitas_examples/papers/perq/benchmark-rotation_block_size.yml
src/brevitas_examples/papers/perq/benchmark.py
src/brevitas_examples/papers/perq/llama3-perq_dag-int4.yml
src/brevitas_examples/papers/perq/llama3-perq_star-int4.yml
src/brevitas_examples/papers/qronos/README.md
src/brevitas_examples/papers/qronos/llama3-gguf-q4_0.yml
src/brevitas_examples/papers/qronos/llama3-w2-hip-magr.yml
src/brevitas_examples/papers/qronos/llama3-w4-base.yml
src/brevitas_examples/papers/qronos/llama3-w4a4-int-quarot.yml
src/brevitas_examples/papers/qronos/llama3-w4a4-int-spinquant.yml
src/brevitas_examples/papers/qronos/llama3-w4a4-mxfp-quarot.yml
src/brevitas_examples/papers/qronos/llama3-w4a4-mxfp-spinquant.yml
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/main.py
src/brevitas_examples/stable_diffusion/stable_diffusion_args.py
src/brevitas_examples/stable_diffusion/benchmark/benchmark_template.yaml
src/brevitas_examples/stable_diffusion/benchmark/diffusion_benchmark.py
src/brevitas_examples/stable_diffusion/config/default_template.yaml
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