Py2D is a Python solver for incompressible 2-dimensional (2D) Navier-stokes equations. Py2D leverages JAX, a high-performance numerical computing library, allowing for rapid execution on GPUs while ...