Yes! Via lifecycle hooks - this memory consumption example is similar in principle https://kedro.readthedocs.io/en/latest/07_extend_kedro/02_hooks.html#add-memory-consumption-tracking