.gitignore
AGENTS.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
pyrightconfig.json
uv.lock
.github/workflows/publish-to-pypi.yml
tests/__init__.py
tests/dependency_graph.py
tests/fixtures_models.py
tests/test_attention_axes_e2e.py
tests/test_attention_axes_unit.py
tests/test_calculation_runtime.py
tests/test_calculation_specs.py
tests/test_group_lasso_regularizer.py
tests/test_l2_norm_distribution_tracker.py
tests/test_linear_weight_sum_specs.py
tests/test_mha_operations.py
tests/test_model_structured_sparsity.py
tests/test_module_bitrate_extractor.py
tests/test_nvidia_2_4_sparsity_tracker.py
tests/test_param_pr_unit.py
tests/test_param_unit_calculator.py
tests/test_reduction_plan_builder.py
tests/test_sparsity_controller.py
tests/test_weight_tracker_unit_extensive.py
tests/test_weighttracker_fvcore_e2e.py
torch_weighttracker/__init__.py
torch_weighttracker/canonical_units.py
torch_weighttracker/consumer_ignore.py
torch_weighttracker/py.typed
torch_weighttracker/weight_tracker.py
torch_weighttracker.egg-info/PKG-INFO
torch_weighttracker.egg-info/SOURCES.txt
torch_weighttracker.egg-info/dependency_links.txt
torch_weighttracker.egg-info/requires.txt
torch_weighttracker.egg-info/top_level.txt
torch_weighttracker/calculations/__init__.py
torch_weighttracker/calculations/base.py
torch_weighttracker/calculations/cached_calc.py
torch_weighttracker/calculations/calculations.py
torch_weighttracker/calculations/context.py
torch_weighttracker/calculations/pipeline_calc.py
torch_weighttracker/calculations/reduction_calc.py
torch_weighttracker/calculations/registry.py
torch_weighttracker/calculations/spec.py
torch_weighttracker/calculations/static_calc.py
torch_weighttracker/calculations/calcs/__init__.py
torch_weighttracker/calculations/calcs/active_macs_pr_module.py
torch_weighttracker/calculations/calcs/active_units.py
torch_weighttracker/calculations/calcs/baseline_group_sizes.py
torch_weighttracker/calculations/calcs/baseline_macs_pr_module.py
torch_weighttracker/calculations/calcs/baseline_module_axes.py
torch_weighttracker/calculations/calcs/baseline_param_pr_unit_pr_group.py
torch_weighttracker/calculations/calcs/bitrate_pr_module.py
torch_weighttracker/calculations/calcs/block_2_4_sparsity.py
torch_weighttracker/calculations/calcs/group_change_effect.py
torch_weighttracker/calculations/calcs/group_sizes.py
torch_weighttracker/calculations/calcs/group_unit_param_change.py
torch_weighttracker/calculations/calcs/groups_to_units.py
torch_weighttracker/calculations/calcs/l2_norm_pr_unit.py
torch_weighttracker/calculations/calcs/module_axis_cost_indices.py
torch_weighttracker/calculations/calcs/param_pr_unit.py
torch_weighttracker/calculations/calcs/structured_unit_sum.py
torch_weighttracker/calculations/calcs/unit_active_mask.py
torch_weighttracker/calculations/calcs/unit_delta_to_module_axis.py
torch_weighttracker/calculations/calcs/units_to_group.py
torch_weighttracker/calculations/calcs/units_to_module_axis.py
torch_weighttracker/calculations/calcs/unstructured_sparsity_pr_module.py
torch_weighttracker/extractors/codeq_bitrate_extractor.py
torch_weighttracker/extractors/extractor.py
torch_weighttracker/extractors/parameter_extractor.py
torch_weighttracker/operations/__init__.py
torch_weighttracker/operations/base.py
torch_weighttracker/operations/conv.py
torch_weighttracker/operations/generic.py
torch_weighttracker/operations/linear.py
torch_weighttracker/operations/mha.py
torch_weighttracker/operations/norm.py
torch_weighttracker/operations/resolver.py
torch_weighttracker/plans/mapping_plan.py
torch_weighttracker/plans/module_axis_plan.py
torch_weighttracker/plans/unit_weight_operation_plan.py
torch_weighttracker/reductions/builder.py
torch_weighttracker/reductions/compiler.py
torch_weighttracker/reductions/helpers.py
torch_weighttracker/reductions/ops.py
torch_weighttracker/regularizers/__init__.py
torch_weighttracker/regularizers/base.py
torch_weighttracker/regularizers/group_lasso.py
torch_weighttracker/regularizers/group_lasso_with_bitrate.py
torch_weighttracker/torch_pruning/__init__.py
torch_weighttracker/torch_pruning/_helpers.py
torch_weighttracker/torch_pruning/ops.py
torch_weighttracker/torch_pruning/dependency/__init__.py
torch_weighttracker/torch_pruning/dependency/constants.py
torch_weighttracker/torch_pruning/dependency/dependency.py
torch_weighttracker/torch_pruning/dependency/graph.py
torch_weighttracker/torch_pruning/dependency/group.py
torch_weighttracker/torch_pruning/dependency/index_mapping.py
torch_weighttracker/torch_pruning/dependency/node.py
torch_weighttracker/torch_pruning/dependency/shape_infer.py
torch_weighttracker/torch_pruning/pruner/__init__.py
torch_weighttracker/torch_pruning/pruner/function.py
torch_weighttracker/torch_pruning/utils/__init__.py
torch_weighttracker/torch_pruning/utils/utils.py
torch_weighttracker/trackers/__init__.py
torch_weighttracker/trackers/base.py
torch_weighttracker/trackers/group_names.py
torch_weighttracker/trackers/group_pruning_summary.py
torch_weighttracker/trackers/l2_norm_distribution.py
torch_weighttracker/trackers/nvidia_2_4_sparsity.py
torch_weighttracker/trackers/structured_bops.py
torch_weighttracker/trackers/unstructured_sparsity.py