Any example notebook for Rayleigh-Benard convection ?

Hi,

I was going through this paper ["Toward accelerated data-driven Rayleigh–Bénard convection simulations"](https://link.springer.com/article/10.1140/epje/s10189-023-00302-w) which used jax-cfd and some of the co-authors are developers of this library. I was wondering if you have a notebook for 2D Rayleigh-benard convection.

Thanks,
Andre