This is a much needed feature for developing  pytorch models in jupyter. This issue is preventing incremental development which is the essence of notebooks so that there is no need to re execute the entire code.