Request: Example notebook showing training of a hybrid model

[akasharidas]

The readme says that an example notebook showing how to train a simple hybrid ML + CFD model is being prepared. Is there an ETA on this? I'm trying to get it to work using JAX-CFD and Flax but I'm running into problems. An example notebook would help me a lot!