#7 implemented basic differentiability, but currently we get NaN gradients in these tests.

• Investigate and fix the NaN gradients (Track down differentiability issues  #25)
• Implement finite differences comparison for the time stepping was well
• Use and test checkpointing with Terrarium