| """DAVINCI SparseMoE TENP pruning: signed projection, gate^1, modality z-score.""" |
|
|
| from .apply import prune_moe_layer |
| from .collector import ExpertStats, LayerCollector |
| from .hook import attach_collectors, detach_collectors, maybe_attach_from_env, maybe_save_from_env, set_collectors_timestep |
| from .score import gated_signed_projection |
| from .scoring import fuse_modality_scores, keep_indices, modality_lambdas |
| from .sttenp import prune_sttenp_layer |
|
|
| __all__ = [ |
| "ExpertStats", |
| "LayerCollector", |
| "attach_collectors", |
| "detach_collectors", |
| "maybe_attach_from_env", |
| "maybe_save_from_env", |
| "set_collectors_timestep", |
| "fuse_modality_scores", |
| "gated_signed_projection", |
| "keep_indices", |
| "modality_lambdas", |
| "prune_moe_layer", |
| "prune_sttenp_layer", |
| ] |
|
|
