You should checkout lbmpy for generation of lbm kernels: https://pycodegen.pages.i10git.cs.fau.de/lbmpy/