Is there a design for ModeInfo anywhere? I suppose I'll find out more about what it's used for as I read this PR, but the comment here doesn't really help (and we didn't have anything like ModeInfo for torch function)