Agree with that. The most elegant convenience feature would be to fixate nn.modules after an auto-trace.
AFAIK tracing is already working well for ONNX export, is it? Auto-tracing should work with yet undimensioned nn.Modules (and with the usual caveats regarding stochastic control-flow).