Running jax-cfd on a Custom Mesh

[cli651]

Hi,

Apologies if this is a bad question, but is it possible to run jax-cfd on a custom 2D mesh - even a simple one, say a channel with an area expansion? After some digging through the repo, it seems that this can only be run on a purely rectangular grid, though I could be wrong - any guidance would be greatly appreciated.