Of course. But that somehow defeats the purpose of progressive disclosure of complexity IMO. I wanted to able to focus on my training loop and delegate rest of the things to the framework whenever possible.
And the link does not elaboratively suggest a workaround for the use-case I mentioned.