Py_SWE This project contains a solver for the shallow water equations, implemented in Python using JAX to enable the differentiation of the solver.