I gave a small talk on how to really push using hooks for logging intermediate values (including capturing gradients from torch & fx scripted modules) that may

I gave a small talk on how to really push using hooks for logging intermediate values (including capturing gradients from torch & fx scripted modules) that may be useful: https://static.sched.com/hosted_files/pytorch2023/40/Intermediate%20Logging%20_%20PyTorch%20Conference%202023.pdf https://static.sched.com/hosted_files/pytorch2023/40/Interme...