Activation hooks redesign (reuse hooks component across both minitron and puzzletron)#1022

Open

danielkorzekwa wants to merge 9 commits intomainfrom

dkorzewa/activation_hooks_redesign_minitron_puzzletron

Commits on Mar 13, 2026

Add shared activation hooks infrastructure for minitron and puzzletron
danielkorzekwa
committed
Add logger dependency for activation hooks
danielkorzekwa
committed
Add a logger and json_dump needed by base_hooks (TODO: both should be moved later outside of puzzletron module)
danielkorzekwa
committed
Fix a broken test due to a missing random seed.
danielkorzekwa
committed
Logging and robust_json refactoring.
danielkorzekwa
committed
Delete not needed logger and robust_json (moved to modelopt)
danielkorzekwa
committed
Change the location for activation hooks from nas..... to prune.importance_hooks
danielkorzekwa
committed
Add a security comment why using torch.load(weight_only=False)
danielkorzekwa
committed
No need for computing gradients during a loop for activation hooks analysis
danielkorzekwa
committed