Hooks for Flux or Lux

I just discovered how to use “Hooks” in Pytorch, which allows me to accumulate statistics and any other information in the different layers of my network (e.g., activation, weights, etc.). Is there a similar approach to collecting diagnostics when using Flux or Lux? Hooks is a form of callback. Thanks.