pruning
=======

.. List the submodules



.. rubric:: Modules

.. autosummary::
   :toctree:
   :recursive:



   modelopt.torch.puzzletron.pruning.attention_ffn_surgery



   modelopt.torch.puzzletron.pruning.compact_runtime



   modelopt.torch.puzzletron.pruning.dynamic_block_prune



   modelopt.torch.puzzletron.pruning.elastic_sampling



   modelopt.torch.puzzletron.pruning.embedding_pruning



   modelopt.torch.puzzletron.pruning.expert_removal_pruning_mixin



   modelopt.torch.puzzletron.pruning.ffn_intermediate_pruning_mixin



   modelopt.torch.puzzletron.pruning.gated_delta_net



   modelopt.torch.puzzletron.pruning.gated_delta_net_pruning_mixin



   modelopt.torch.puzzletron.pruning.kv_heads_pruning_mixin



   modelopt.torch.puzzletron.pruning.latent_moe_surgery



   modelopt.torch.puzzletron.pruning.mamba2_surgery



   modelopt.torch.puzzletron.pruning.materialize



   modelopt.torch.puzzletron.pruning.moe_mamba_pruning_mixin



   modelopt.torch.puzzletron.pruning.ple_pruning



   modelopt.torch.puzzletron.pruning.pruning_mixin



   modelopt.torch.puzzletron.pruning.pruning_utils



   modelopt.torch.puzzletron.pruning.runtime_candidate



   modelopt.torch.puzzletron.pruning.runtime_hidden_width



   modelopt.torch.puzzletron.pruning.runtime_ple



   modelopt.torch.puzzletron.pruning.sorted_teacher




.. Autodoc anything defined in the module itself

   TODO: WE DON'T USE THIS OPTION RIGHT NOW BUT WE CAN REACTIVATE IF WANTED
   We use :ignore-module-all: so sphinx does not document the same module twice, even if it is reimported
   For reimports that should be documented somewhere other than where they are defined, the re-imports
   __module__ should be manually overridden -- i.e. in the ``__init__.py`` which contains ``from xxx import YYY``,
   add in ``YYY.__module__ = __name__``.

.. automodule:: modelopt.torch.puzzletron.pruning
   :members:
   :undoc-members:

   .. Also show members without docstrings. Only members from __all__ are considered as per conf.py
   .. Ideally we should add docstrings for these members.


   .. Overview table of available classes in the module
   
   
   


   .. Overview table of available functions in the module