This is awesome, thanks so much for sharing this - and I love how whimsical the docs are: Our pond is a perfect 500 x 500 square, as is the case for most ponds

This is awesome, thanks so much for sharing this - and I love how whimsical the docs are: Our pond is a perfect 500 x 500 square, as is the case for most ponds found in nature. A feedback on the PDE aspect: is there any nice way to write your model and grid using TensorFlow, and still obtain the free derivatives/optimized computation graph if you use a different solver to actually integrate the equations? The example shows off a very simple toy first order Euler solver, but in practice this is grossly inadequate. I suspect that it's very difficult if not impossible - especially since most solver libraries are very old Fortran libs, and carrying forward the reverse mode differentiation there is a nightmare. Still, this makes for very nice building blocks. Another question - would you consider adding an example of using this together with Google Ceres - to make the two autodiffs play nice together?