Makani: Massively parallel AI based weather and climate models
github.com