Update the documentation of the models

[amal-ghamdi]

Description:

The documentation of models need some update to help the user know for example how to define a non-linear model, pass gradient, Jacobian, etc

DoD:

• [ ] The updates required is decided upon
• [ ] The documentation is updated, reviewed and merged