[Tasks]: Add a notebook showing how differentiable versions of two module work

[SarahAlidoost]

After #8, create a notebook that shows how two modules work. The code in tests can be used in the notebook. Add the notebook to the documentation and setup doc page.