neuralgcm
Coordinate axes for scientific computing in JAX
NeuralGCM's differentiable dycore
Hybrid ML + physics model of the Earth's atmosphere