Wuyang

How to get the simulated pressure variable along the trajectory? For example in this notebook: https://github.com/google/jax-cfd/blob/main/notebooks/demo.ipynb Thank you very much!

Dear researchers: Do you happen to have any example of using [kolmogorov_forcing](https://github.com/google/jax-cfd/blob/main/jax_cfd/base/forcings.py#L63) in [semi_implicit_navier_stokes](https://github.com/google/jax-cfd/blob/main/jax_cfd/base/equations.py#L128) ? Naively passing `kolmogorov_forcing` into `semi_implicit_navier_stokes` will not work. Mimic the usage in [spectral](https://github.com/google/jax-cfd/blob/main/jax_cfd/spectral/equations.py#L213) will give...