.gitattributes
.gitignore
LICENSE
MANIFEST.in
README.md
VERSION
compile_docker_requirements.sh
pyproject.toml
release.sh
requirements-dev.txt
requirements.txt
setup.py
.github/dependabot.yml
.github/workflows/heavy-tests-docker.yml
.github/workflows/heavy-tests-env.yml
.github/workflows/light-tests-docker.yml
.github/workflows/light-tests-env.yml
.github/workflows/release.yml
dev_tools/__init__.py
dev_tools/algorithms_testing_tools.py
dev_tools/constants.py
dev_tools/model_loading.py
dev_tools/model_size.json
dev_tools/model_test_results.json
dev_tools/model_test_results_v0.json
dev_tools/supported_models.ipynb
dev_tools/test_model.py
dev_tools/testing.py
dev_tools/testing_result_registry.py
dev_tools/timeout.py
dev_tools/update_model_size.py
docker/Dockerfile
docker/build_docker_image_locally.sh
docker/requirements-docker.txt
node_api_conversion/__init__.py
node_api_conversion/convert_cell_to_dag_module.py
node_api_conversion/convert_dag_module_to_cell.py
node_api_conversion/from_nd_converter.py
node_api_conversion/to_nd_converter.py
node_api_conversion/utils.py
node_api_conversion/tests/__init__.py
node_api_conversion/tests/test_conversion_from_nd.py
resources/channel_pruning_supported_models_table.md
resources/conversion_to_dag_module.md
resources/pruning_plots.ipynb
resources/pruning_readme.md
resources/pruning_results.md
resources/supported_models_table.md
resources/examples/mnist_notebook.ipynb
resources/examples/custom_models/swiftformer_example.ipynb
resources/examples/custom_models/swiftformer_example.py
resources/pruning_results_plots/f1.png
resources/pruning_results_plots/f2.png
resources/pruning_results_plots/f3.png
resources/pruning_results_plots/f4.png
resources/pruning_results_plots/f5.png
tests/__init__.py
tests/conftest.py
tests/test_constants.py
tests/test_custom_module_handling.py
tests/test_dag_compile.py
tests/test_flops_computation.py
tests/test_load_optimized.py
tests/test_regular_tracing.py
tests/test_serialization.py
tests/test_timm.py
tests/test_timm_extended_orbitalization.py
tests/test_torch_compile.py
tests/test_utils.py
tests/commons/__init__.py
tests/commons/test_params_computation.py
tests/dag_from_unstructured/__init__.py
tests/dag_from_unstructured/test_conversion_to_dag.py
tests/graph_surgery/__init__.py
tests/graph_surgery/test_simple_graph_surgery.py
tests/orbits/__init__.py
torch_dag/__init__.py
torch_dag/structured_modules.py
torch_dag.egg-info/PKG-INFO
torch_dag.egg-info/SOURCES.txt
torch_dag.egg-info/dependency_links.txt
torch_dag.egg-info/entry_points.txt
torch_dag.egg-info/requires.txt
torch_dag.egg-info/top_level.txt
torch_dag.egg-info/zip-safe
torch_dag/api/__init__.py
torch_dag/api/commons.py
torch_dag/api/core.py
torch_dag/api/io.py
torch_dag/api/patterns.py
torch_dag/api/patterns_v2.py
torch_dag/api/visualization.py
torch_dag/builders/__init__.py
torch_dag/commons/__init__.py
torch_dag/commons/compute_stats.py
torch_dag/commons/dag_processors.py
torch_dag/commons/debugging_tools.py
torch_dag/commons/flops_computation.py
torch_dag/commons/mixins.py
torch_dag/conversion/export_onnx.py
torch_dag/core/__init__.py
torch_dag/core/common_tools.py
torch_dag/core/dag_module.py
torch_dag/core/dag_module_utils.py
torch_dag/core/dag_tracer.py
torch_dag/core/module_handling.py
torch_dag/core/module_handling_for_pruning.py
torch_dag/core/prunable_modules.py
torch_dag/core/unstructured_to_structured.py
torch_dag/io/__init__.py
torch_dag/io/load_dag_from_path.py
torch_dag/patterns/__init__.py
torch_dag/patterns/crude_patterns.py
torch_dag/patterns/specific_patterns.py
torch_dag/patterns_v2/__init__.py
torch_dag/patterns_v2/match.py
torch_dag/patterns_v2/pattern.py
torch_dag/patterns_v2/pattern_node.py
torch_dag/patterns_v2/transform.py
torch_dag/visualization/__init__.py
torch_dag/visualization/visualize_dag.py
torch_dag_algorithms/__init__.py
torch_dag_algorithms/cli.py
torch_dag_algorithms/commons/__init__.py
torch_dag_algorithms/commons/flops_computation.py
torch_dag_algorithms/commons/params_computation.py
torch_dag_algorithms/pruning/README.md
torch_dag_algorithms/pruning/__init__.py
torch_dag_algorithms/pruning/channel_removal_interface.py
torch_dag_algorithms/pruning/channel_removal_primitives.py
torch_dag_algorithms/pruning/commons.py
torch_dag_algorithms/pruning/constants.py
torch_dag_algorithms/pruning/dag_orbitalizer.py
torch_dag_algorithms/pruning/filters.py
torch_dag_algorithms/pruning/mask_propagation.py
torch_dag_algorithms/pruning/masking_inserter.py
torch_dag_algorithms/pruning/masking_insertion_strategy.py
torch_dag_algorithms/pruning/module_multipliers.py
torch_dag_algorithms/pruning/modules.py
torch_dag_algorithms/pruning/orbit.py
torch_dag_algorithms/pruning/orbitalize_model.py
torch_dag_algorithms/pruning/orbits_extraction.py
torch_dag_algorithms/pruning/orbits_search_stage.py
torch_dag_algorithms/pruning/plugin_tools.py
torch_dag_algorithms/pruning/prune_channels.py
torch_dag_algorithms/pruning/remove_channels.py
torch_dag_algorithms/pruning/utils.py
torch_dag_algorithms/pruning/losses/__init__.py
torch_dag_algorithms/pruning/losses/bkd_losses.py
torch_dag_algorithms/pruning/losses/entropy_losses.py
torch_dag_algorithms/pruning/losses/latency_losses.py
torch_dag_algorithms/pruning/losses/multiplier.py
torch_dag_timm_plugin/__init__.py
torch_dag_timm_plugin/channel_removal.py
torch_dag_timm_plugin/constants.py
torch_dag_timm_plugin/mask_propagation.py
torch_dag_timm_plugin/modified_timm_modules.py
torch_dag_timm_plugin/module_conversion.py
torch_dag_timm_plugin/module_multipliers.py
torch_dag_timm_plugin/timm_flops_computation.py
torch_dag_timm_plugin/timm_modules_handling.py