Maybe similar to this (when using streamfunction-vorticity again): https://github.com/Ceyron/machine-learning-and-simulation/blob/main/english/simulation_scripts/taylor_green_turbulence.jl