Probably it'll fit well in GeophysicalFlows.jl A notebook I've put together at some point is [here](https://github.com/navidcy/CLExWinterSchool2020/tree/master/simulations/SWE). This was based on the example of linear SWE that exists in [FourierFlows.jl](https://github.com/FourierFlows/FourierFlows.jl). @glwagner, @apaloczy