LICENSE
README.md
README.pypi.md
pyproject.toml
src/coreai_opt/__init__.py
src/coreai_opt/_about.py
src/coreai_opt/base_model_compressor.py
src/coreai_opt/common.py
src/coreai_opt/py.typed
src/coreai_opt.egg-info/PKG-INFO
src/coreai_opt.egg-info/SOURCES.txt
src/coreai_opt.egg-info/dependency_links.txt
src/coreai_opt.egg-info/requires.txt
src/coreai_opt.egg-info/top_level.txt
src/coreai_opt/_utils/api_visibility_utils.py
src/coreai_opt/_utils/casting_utils.py
src/coreai_opt/_utils/config_utils.py
src/coreai_opt/_utils/eager_utils.py
src/coreai_opt/_utils/export_utils.py
src/coreai_opt/_utils/import_utils.py
src/coreai_opt/_utils/metadata_utils.py
src/coreai_opt/_utils/python_utils.py
src/coreai_opt/_utils/registry_utils.py
src/coreai_opt/_utils/repo_utils.py
src/coreai_opt/_utils/spec_utils.py
src/coreai_opt/_utils/torch_utils.py
src/coreai_opt/_utils/version_utils.py
src/coreai_opt/_utils/insertion/torch_function/__init__.py
src/coreai_opt/_utils/insertion/torch_function/base_supported_ops_registry.py
src/coreai_opt/_utils/insertion/torch_function/handler.py
src/coreai_opt/_utils/insertion/torch_function/modes.py
src/coreai_opt/_utils/insertion/torch_function/module_boundary_tracker.py
src/coreai_opt/_utils/insertion/torch_function/preregistration_tracker.py
src/coreai_opt/_utils/insertion/torch_function/registered_optimizers_tracker.py
src/coreai_opt/_utils/insertion/torch_function/types.py
src/coreai_opt/_utils/insertion/torch_function/utils.py
src/coreai_opt/casting/__init__.py
src/coreai_opt/casting/casting.py
src/coreai_opt/config/__init__.py
src/coreai_opt/config/compression_config.py
src/coreai_opt/config/spec/__init__.py
src/coreai_opt/config/spec/base.py
src/coreai_opt/config/spec/compression_simulator.py
src/coreai_opt/config/spec/factory.py
src/coreai_opt/coreai_utils/__init__.py
src/coreai_opt/coreai_utils/_coreai_imports.py
src/coreai_opt/coreai_utils/common.py
src/coreai_opt/coreai_utils/_utils/__init__.py
src/coreai_opt/coreai_utils/_utils/graph_utils.py
src/coreai_opt/coreai_utils/_utils/palettize_utils.py
src/coreai_opt/coreai_utils/_utils/quantize_utils.py
src/coreai_opt/coreai_utils/_utils/sparse_utils.py
src/coreai_opt/coreai_utils/_utils/type_utils.py
src/coreai_opt/coreai_utils/passes/__init__.py
src/coreai_opt/coreai_utils/passes/weight_palettization.py
src/coreai_opt/coreai_utils/passes/weight_quantization.py
src/coreai_opt/coreai_utils/passes/weight_sparsification.py
src/coreai_opt/inspection/__init__.py
src/coreai_opt/inspection/_formatting.py
src/coreai_opt/inspection/_graph_mode.py
src/coreai_opt/inspection/model_inspector.py
src/coreai_opt/inspection/types.py
src/coreai_opt/palettization/__init__.py
src/coreai_opt/palettization/base_palettizer.py
src/coreai_opt/palettization/config/__init__.py
src/coreai_opt/palettization/config/palettization_config.py
src/coreai_opt/palettization/config/_presets/__init__.py
src/coreai_opt/palettization/config/_presets/kmeans_palettizer_config.py
src/coreai_opt/palettization/config/_presets/module_kmeans_palettizer_config.py
src/coreai_opt/palettization/kmeans/__init__.py
src/coreai_opt/palettization/kmeans/_efficient_kmeans.py
src/coreai_opt/palettization/kmeans/_prepare_for_export.py
src/coreai_opt/palettization/kmeans/kmeans_fake_palettize.py
src/coreai_opt/palettization/kmeans/kmeans_support_mixins.py
src/coreai_opt/palettization/kmeans/palettizer.py
src/coreai_opt/palettization/kmeans/supported_ops_registry.py
src/coreai_opt/palettization/spec/__init__.py
src/coreai_opt/palettization/spec/errors.py
src/coreai_opt/palettization/spec/factory.py
src/coreai_opt/palettization/spec/fake_palettize.py
src/coreai_opt/palettization/spec/granularity.py
src/coreai_opt/palettization/spec/spec.py
src/coreai_opt/pruning/__init__.py
src/coreai_opt/pruning/_prepare_for_export.py
src/coreai_opt/pruning/base_pruner.py
src/coreai_opt/pruning/magnitude_pruner.py
src/coreai_opt/pruning/supported_ops_registry.py
src/coreai_opt/pruning/config/__init__.py
src/coreai_opt/pruning/config/magnitude_pruner_config.py
src/coreai_opt/pruning/config/sparsity_schedule.py
src/coreai_opt/pruning/spec/__init__.py
src/coreai_opt/pruning/spec/prune.py
src/coreai_opt/pruning/spec/scheme.py
src/coreai_opt/pruning/spec/spec.py
src/coreai_opt/quantization/__init__.py
src/coreai_opt/quantization/_axis_defaults.py
src/coreai_opt/quantization/_export_utils.py
src/coreai_opt/quantization/_utils.py
src/coreai_opt/quantization/base_quantizer.py
src/coreai_opt/quantization/quantizer.py
src/coreai_opt/quantization/_eager/__init__.py
src/coreai_opt/quantization/_eager/_prepare_for_export.py
src/coreai_opt/quantization/_eager/_prepare_for_mil_export.py
src/coreai_opt/quantization/_eager/_utils.py
src/coreai_opt/quantization/_eager/quantizer.py
src/coreai_opt/quantization/_eager/supported_ops_registry.py
src/coreai_opt/quantization/_graph/__init__.py
src/coreai_opt/quantization/_graph/_annotation_config.py
src/coreai_opt/quantization/_graph/_annotation_pattern_registry.py
src/coreai_opt/quantization/_graph/_annotation_utils.py
src/coreai_opt/quantization/_graph/_conv_bn_utils.py
src/coreai_opt/quantization/_graph/_prepare_for_export.py
src/coreai_opt/quantization/_graph/_utils.py
src/coreai_opt/quantization/_graph/quantizer.py
src/coreai_opt/quantization/config/__init__.py
src/coreai_opt/quantization/config/quantization_config.py
src/coreai_opt/quantization/config/_presets/__init__.py
src/coreai_opt/quantization/config/_presets/module_quantizer_config.py
src/coreai_opt/quantization/config/_presets/quantizer_config.py
src/coreai_opt/quantization/spec/__init__.py
src/coreai_opt/quantization/spec/errors.py
src/coreai_opt/quantization/spec/factory.py
src/coreai_opt/quantization/spec/fake_quantize.py
src/coreai_opt/quantization/spec/granularity.py
src/coreai_opt/quantization/spec/qformulation.py
src/coreai_opt/quantization/spec/qparams_calculator.py
src/coreai_opt/quantization/spec/qscheme.py
src/coreai_opt/quantization/spec/range_calculator.py
src/coreai_opt/quantization/spec/spec.py
tests/test_api_visibility.py
tests/test_api_visibility_utils.py
tests/test_base_model_compressor.py
tests/test_compression_config.py
tests/test_inspection.py
tests/test_joint_compression.py
tests/test_palettization_preset.py
tests/test_quantization_preset.py
tests/test_smoke.py