LICENSE
README.md
setup.py
machop/chop/__init__.py
machop/chop/cli.py
machop/chop/actions/__init__.py
machop/chop/actions/accelerate_peft.py
machop/chop/actions/accelerate_train.py
machop/chop/actions/test.py
machop/chop/actions/train.py
machop/chop/actions/transform.py
machop/chop/actions/validate.py
machop/chop/actions/search/__init__.py
machop/chop/actions/search/search.py
machop/chop/actions/search/search_space/__init__.py
machop/chop/actions/search/search_space/base.py
machop/chop/actions/search/search_space/utils.py
machop/chop/actions/search/search_space/quantization/__init__.py
machop/chop/actions/search/search_space/quantization/graph.py
machop/chop/actions/search/search_space/quantization/manual_hf_module.py
machop/chop/actions/search/strategies/__init__.py
machop/chop/actions/search/strategies/base.py
machop/chop/actions/search/strategies/optuna.py
machop/chop/actions/search/strategies/rl/__init__.py
machop/chop/actions/search/strategies/rl/core_algorithm.py
machop/chop/actions/search/strategies/rl/env.py
machop/chop/actions/search/strategies/runners/__init__.py
machop/chop/actions/search/strategies/runners/hardware/__init__.py
machop/chop/actions/search/strategies/runners/hardware/base.py
machop/chop/actions/search/strategies/runners/hardware/dummy.py
machop/chop/actions/search/strategies/runners/hardware/memory/__init__.py
machop/chop/actions/search/strategies/runners/hardware/memory/avg_bitwidth.py
machop/chop/actions/search/strategies/runners/hardware/memory/model_profiler.py
machop/chop/actions/search/strategies/runners/software/__init__.py
machop/chop/actions/search/strategies/runners/software/base.py
machop/chop/actions/search/strategies/runners/software/eval.py
machop/chop/actions/search/strategies/runners/software/train.py
machop/chop/dataset/__init__.py
machop/chop/dataset/toy_dataset.py
machop/chop/dataset/utils.py
machop/chop/dataset/nerf/__init__.py
machop/chop/dataset/nerf/blender.py
machop/chop/dataset/nerf/ray_utils.py
machop/chop/dataset/nlp/__init__.py
machop/chop/dataset/nlp/language_modeling.py
machop/chop/dataset/nlp/sentiment_analysis.py
machop/chop/dataset/nlp/text_entailment.py
machop/chop/dataset/nlp/translation.py
machop/chop/dataset/physical/__init__.py
machop/chop/dataset/physical/jsc.py
machop/chop/dataset/vision/__init__.py
machop/chop/dataset/vision/cifar.py
machop/chop/dataset/vision/imagenet.py
machop/chop/dataset/vision/mnist.py
machop/chop/dataset/vision/transform.py
machop/chop/dataset/vision/transforms/__init__.py
machop/chop/dataset/vision/transforms/cifar.py
machop/chop/dataset/vision/transforms/imagenet.py
machop/chop/dataset/vision/transforms/mnist.py
machop/chop/dataset/vision/transforms/presets.py
machop/chop/ir/__init__.py
machop/chop/ir/graph/__init__.py
machop/chop/ir/graph/mase_graph.py
machop/chop/ir/graph/mase_metadata.py
machop/chop/models/__init__.py
machop/chop/models/huggingface_nlp_models.py
machop/chop/models/ops.py
machop/chop/models/utils.py
machop/chop/models/manual/__init__.py
machop/chop/models/manual/lora_modules.py
machop/chop/models/manual/lora_utils.py
machop/chop/models/manual/quant_utils.py
machop/chop/models/manual/sparse_modules.py
machop/chop/models/manual/sparse_utils.py
machop/chop/models/manual/bert_quantized/__init__.py
machop/chop/models/manual/bert_quantized/configuration_bert.py
machop/chop/models/manual/bert_quantized/modeling_bert.py
machop/chop/models/manual/bert_quantized/quant_config_bert.py
machop/chop/models/manual/llama_llora/__init__.py
machop/chop/models/manual/llama_llora/configuration_llama_llora.py
machop/chop/models/manual/llama_llora/lora_config_llama.py
machop/chop/models/manual/llama_llora/modeling_llama_llora.py
machop/chop/models/manual/llama_plain/__init__.py
machop/chop/models/manual/llama_plain/configuration_llama.py
machop/chop/models/manual/llama_plain/modeling_llama.py
machop/chop/models/manual/llama_quantized/__init__.py
machop/chop/models/manual/llama_quantized/configuration_llama.py
machop/chop/models/manual/llama_quantized/modeling_llama.py
machop/chop/models/manual/llama_quantized/quant_config_llama.py
machop/chop/models/manual/llama_sparse/__init__.py
machop/chop/models/manual/llama_sparse/configuration_llama_sparse.py
machop/chop/models/manual/llama_sparse/modeling_llama_sparse.py
machop/chop/models/manual/llama_sparse/sparse_config_llama.py
machop/chop/models/manual/opt_lora/__init__.py
machop/chop/models/manual/opt_lora/configuration_opt_lora.py
machop/chop/models/manual/opt_lora/lora_config_opt.py
machop/chop/models/manual/opt_lora/modeling_opt_lora.py
machop/chop/models/manual/opt_lora/utils_opt.py
machop/chop/models/manual/opt_plain/__init__.py
machop/chop/models/manual/opt_plain/configuration_opt.py
machop/chop/models/manual/opt_plain/modeling_opt.py
machop/chop/models/manual/opt_quantized/__init__.py
machop/chop/models/manual/opt_quantized/configuration_opt.py
machop/chop/models/manual/opt_quantized/modeling_opt.py
machop/chop/models/manual/opt_quantized/quant_config_opt.py
machop/chop/models/manual/opt_sparse/__init__.py
machop/chop/models/manual/opt_sparse/configuration_opt_sparse.py
machop/chop/models/manual/opt_sparse/modeling_opt_sparse.py
machop/chop/models/manual/opt_sparse/sparse_config_opt.py
machop/chop/models/manual/opt_sparse/utils_opt.py
machop/chop/models/nerf/__init__.py
machop/chop/models/nerf/nerf/__init__.py
machop/chop/models/nerf/nerf/layers.py
machop/chop/models/nerf/nerf/rendering.py
machop/chop/models/patched/__init__.py
machop/chop/models/patched/custom_nlp_modules.py
machop/chop/models/patched/patched_nlp_models.py
machop/chop/models/patched/bert_patched/__init__.py
machop/chop/models/patched/bert_patched/configuration_bert_patched.py
machop/chop/models/patched/bert_patched/modeling_bert_patched.py
machop/chop/models/patched/bert_patched/utils_bert_patched.py
machop/chop/models/patched/opt_patched/__init__.py
machop/chop/models/patched/opt_patched/configuration_opt_patched.py
machop/chop/models/patched/opt_patched/modeling_opt_patched.py
machop/chop/models/patched/opt_patched/utils_opt_patched.py
machop/chop/models/physical/__init__.py
machop/chop/models/physical/jet_substructure/__init__.py
machop/chop/models/toys/__init__.py
machop/chop/models/toys/toy.py
machop/chop/models/toys/toy_custom_fn.py
machop/chop/models/vision/__init__.py
machop/chop/models/vision/cnv/__init__.py
machop/chop/models/vision/cnv/cnv.py
machop/chop/models/vision/cswin/__init__.py
machop/chop/models/vision/cswin/cswintransformer.py
machop/chop/models/vision/deit/__init__.py
machop/chop/models/vision/deit/deit.py
machop/chop/models/vision/deit/deit_v2.py
machop/chop/models/vision/efficientnet/__init__.py
machop/chop/models/vision/efficientnet/efficientnet.py
machop/chop/models/vision/lfc/__init__.py
machop/chop/models/vision/lfc/lfc.py
machop/chop/models/vision/mobilenet_v2/__init__.py
machop/chop/models/vision/mobilenet_v2/mobilenet_v2.py
machop/chop/models/vision/mobilenet_v3/__init__.py
machop/chop/models/vision/mobilenet_v3/mobilenetv3.py
machop/chop/models/vision/nerf/__init__.py
machop/chop/models/vision/nerf/nerf.py
machop/chop/models/vision/pvt/__init__.py
machop/chop/models/vision/pvt/pvt.py
machop/chop/models/vision/pvt/pvt_v2.py
machop/chop/models/vision/repvgg/__init__.py
machop/chop/models/vision/repvgg/repvgg.py
machop/chop/models/vision/repvgg/se_block.py
machop/chop/models/vision/resnet/__init__.py
machop/chop/models/vision/resnet/resnet.py
machop/chop/models/vision/vgg_cifar/__init__.py
machop/chop/models/vision/vgg_cifar/vgg_cifar.py
machop/chop/models/vision/vgg_cifar/vgg_orig.py
machop/chop/models/vision/wideresnet/__init__.py
machop/chop/models/vision/wideresnet/wideresnet.py
machop/chop/passes/__init__.py
machop/chop/passes/graph/__init__.py
machop/chop/passes/graph/common.py
machop/chop/passes/graph/utils.py
machop/chop/passes/graph/analysis/__init__.py
machop/chop/passes/graph/analysis/init_metadata.py
machop/chop/passes/graph/analysis/utils.py
machop/chop/passes/graph/analysis/add_metadata/__init__.py
machop/chop/passes/graph/analysis/add_metadata/add_common_metadata.py
machop/chop/passes/graph/analysis/add_metadata/add_hardware_metadata.py
machop/chop/passes/graph/analysis/add_metadata/add_software_metadata.py
machop/chop/passes/graph/analysis/add_metadata/common_metadata_layers.py
machop/chop/passes/graph/analysis/add_metadata/hardware_metadata_layers.py
machop/chop/passes/graph/analysis/add_metadata/software_metadata_layers.py
machop/chop/passes/graph/analysis/flop_estimator/__init__.py
machop/chop/passes/graph/analysis/flop_estimator/deepspeed.py
machop/chop/passes/graph/analysis/flop_estimator/fine_grained.py
machop/chop/passes/graph/analysis/flop_estimator/fx_profiler.py
machop/chop/passes/graph/analysis/flop_estimator/calculator/__init__.py
machop/chop/passes/graph/analysis/flop_estimator/calculator/calc_funcs.py
machop/chop/passes/graph/analysis/flop_estimator/calculator/calc_modules.py
machop/chop/passes/graph/analysis/hardware/__init__.py
machop/chop/passes/graph/analysis/hardware/mase_hardware_evaluator.py
machop/chop/passes/graph/analysis/hardware/parse_synth_rpt.py
machop/chop/passes/graph/analysis/hardware/run_cosim.py
machop/chop/passes/graph/analysis/pruning/__init__.py
machop/chop/passes/graph/analysis/pruning/calculate_natural_sparsity.py
machop/chop/passes/graph/analysis/pruning/calculate_sparsity.py
machop/chop/passes/graph/analysis/pruning/hook_inspector.py
machop/chop/passes/graph/analysis/quantization/__init__.py
machop/chop/passes/graph/analysis/quantization/calculate_avg_bits.py
machop/chop/passes/graph/analysis/report/__init__.py
machop/chop/passes/graph/analysis/report/report_graph.py
machop/chop/passes/graph/analysis/report/report_node.py
machop/chop/passes/graph/analysis/statistical_profiler/__init__.py
machop/chop/passes/graph/analysis/statistical_profiler/profile_statistics.py
machop/chop/passes/graph/analysis/statistical_profiler/stat.py
machop/chop/passes/graph/analysis/statistical_profiler/utils.py
machop/chop/passes/graph/analysis/verify/__init__.py
machop/chop/passes/graph/analysis/verify/common_metadata_layers.py
machop/chop/passes/graph/analysis/verify/hardware_metadata_layers.py
machop/chop/passes/graph/analysis/verify/verify.py
machop/chop/passes/graph/interface/__init__.py
machop/chop/passes/graph/interface/save_and_load.py
machop/chop/passes/graph/patching/__init__.py
machop/chop/passes/graph/patching/mase_op_wrapper.py
machop/chop/passes/graph/transforms/__init__.py
machop/chop/passes/graph/transforms/dse/__init__.py
machop/chop/passes/graph/transforms/dse/partition_to_multi_device.py
machop/chop/passes/graph/transforms/dse/run_dse.py
machop/chop/passes/graph/transforms/pruning/__init__.py
machop/chop/passes/graph/transforms/pruning/load.py
machop/chop/passes/graph/transforms/pruning/prune.py
machop/chop/passes/graph/transforms/pruning/prune_detach_hook.py
machop/chop/passes/graph/transforms/pruning/pruning_methods.py
machop/chop/passes/graph/transforms/pruning/sparse_parameterization.py
machop/chop/passes/graph/transforms/quantize/__init__.py
machop/chop/passes/graph/transforms/quantize/modify.py
machop/chop/passes/graph/transforms/quantize/quantize.py
machop/chop/passes/graph/transforms/quantize/summary.py
machop/chop/passes/graph/transforms/quantize/quant_parsers/__init__.py
machop/chop/passes/graph/transforms/quantize/quant_parsers/parse_quant_config.py
machop/chop/passes/graph/transforms/quantize/quant_parsers/update_node_meta.py
machop/chop/passes/graph/transforms/quantize/quant_parsers/utils.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/__init__.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/add.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/matmul.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/mult.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/relu.py
machop/chop/passes/graph/transforms/quantize/quantized_funcs/sub.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/__init__.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/conv1d.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/conv2d.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/linear.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/pool2d.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/relu.py
machop/chop/passes/graph/transforms/quantize/quantized_modules/utils.py
machop/chop/passes/graph/transforms/quantize/quantizers/__init__.py
machop/chop/passes/graph/transforms/quantize/quantizers/binary.py
machop/chop/passes/graph/transforms/quantize/quantizers/block_fp.py
machop/chop/passes/graph/transforms/quantize/quantizers/block_log.py
machop/chop/passes/graph/transforms/quantize/quantizers/block_minifloat.py
machop/chop/passes/graph/transforms/quantize/quantizers/integer.py
machop/chop/passes/graph/transforms/quantize/quantizers/log.py
machop/chop/passes/graph/transforms/quantize/quantizers/minifloat.py
machop/chop/passes/graph/transforms/quantize/quantizers/quantizers_for_hw.py
machop/chop/passes/graph/transforms/quantize/quantizers/ternary.py
machop/chop/passes/graph/transforms/quantize/quantizers/utils.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/BaseInitializer.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/BaseTrainer.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/MaskBase.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/__init__.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/utils/BinarizeSign.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/utils/__init__.py
machop/chop/passes/graph/transforms/quantize/quantizers/LUTNet/utils/truth_table.py
machop/chop/passes/graph/transforms/quantize/quantizers/LogicNets/__init__.py
machop/chop/passes/graph/transforms/quantize/quantizers/LogicNets/utils.py
machop/chop/passes/graph/transforms/utils/__init__.py
machop/chop/passes/graph/transforms/utils/conv_bn_fusion.py
machop/chop/passes/graph/transforms/utils/extract_subgraph.py
machop/chop/passes/graph/transforms/utils/extract_subgraph_pass.py
machop/chop/passes/graph/transforms/utils/logicnets_fusion.py
machop/chop/passes/graph/transforms/utils/onnx_annotator.py
machop/chop/passes/graph/transforms/utils/remove_assert.py
machop/chop/passes/graph/transforms/utils/remove_dropout.py
machop/chop/passes/graph/transforms/utils/remove_nonsynthesizable_nodes.py
machop/chop/passes/graph/transforms/utils/report.py
machop/chop/passes/graph/transforms/utils/verify.py
machop/chop/passes/graph/transforms/verilog/__init__.py
machop/chop/passes/graph/transforms/verilog/emit_bram.py
machop/chop/passes/graph/transforms/verilog/emit_hls.py
machop/chop/passes/graph/transforms/verilog/emit_internal.py
machop/chop/passes/graph/transforms/verilog/emit_logicnets.py
machop/chop/passes/graph/transforms/verilog/emit_tb.py
machop/chop/passes/graph/transforms/verilog/emit_tb_data_in.py
machop/chop/passes/graph/transforms/verilog/emit_tb_data_out.py
machop/chop/passes/graph/transforms/verilog/emit_tb_testbench.py
machop/chop/passes/graph/transforms/verilog/emit_top.py
machop/chop/passes/graph/transforms/verilog/internal_file_dependences.py
machop/chop/passes/graph/transforms/verilog/logicnets/__init__.py
machop/chop/passes/graph/transforms/verilog/logicnets/bench.py
machop/chop/passes/graph/transforms/verilog/logicnets/emit_linear.py
machop/chop/passes/graph/transforms/verilog/logicnets/util.py
machop/chop/passes/module/__init__.py
machop/chop/passes/module/analysis/__init__.py
machop/chop/passes/module/analysis/quantization/__init__.py
machop/chop/passes/module/analysis/quantization/calculate_avg_bits_module.py
machop/chop/passes/module/transforms/__init__.py
machop/chop/plt_wrapper/__init__.py
machop/chop/plt_wrapper/base.py
machop/chop/plt_wrapper/nerf/__init__.py
machop/chop/plt_wrapper/nerf/losses.py
machop/chop/plt_wrapper/nerf/metrics.py
machop/chop/plt_wrapper/nerf/nerf.py
machop/chop/plt_wrapper/nerf/visualization.py
machop/chop/plt_wrapper/nlp/__init__.py
machop/chop/plt_wrapper/nlp/classification.py
machop/chop/plt_wrapper/nlp/lm.py
machop/chop/plt_wrapper/nlp/translation.py
machop/chop/plt_wrapper/physical/__init__.py
machop/chop/plt_wrapper/physical/jet_substructure.py
machop/chop/plt_wrapper/vision/__init__.py
machop/chop/plt_wrapper/vision/base.py
machop/chop/tools/__init__.py
machop/chop/tools/checkpoint_load.py
machop/chop/tools/config_load.py
machop/chop/tools/get_input.py
machop/chop/tools/logger.py
machop/chop/tools/onnx_importer.py
machop/chop/tools/onnx_utils.py
machop/chop/tools/registry.py
machop/chop/tools/utils.py
machop/mase_tools.egg-info/PKG-INFO
machop/mase_tools.egg-info/SOURCES.txt
machop/mase_tools.egg-info/dependency_links.txt
machop/mase_tools.egg-info/requires.txt
machop/mase_tools.egg-info/top_level.txt