We present a new way of solving the quasi-geostrophic (QG) equations, a simple set of equations describing ocean dynamics. Our method is solely based on the numerical methods used to solve the equations and requires no parameter tuning. Moreover, it can handle non-rectangular geometries, opening the way to study QG equations on realistic domains. We release a PyTorch implementation to ease future machine-learning developments on top of the presented method.