After #49, we want a notebook similar to [09 Optimizing parameters in a PCSE model.ipynb](https://github.com/ajwdewit/pcse_notebooks/blob/master/09%20Optimizing%20parameters%20in%20a%20PCSE%20model.ipynb) using torch optimization workflow. The notebook should be added to the documentation.