Neat use of ChatGPT. Given the usual printout of a PyTorch model (showing arch), you can ask it to infer and generate the python code responsible for that model. I tried simple MLP, no problem. I hope this will extend to more complicated cases, and/or see improvement as it grows