NeuralGCM's differentiable dycore
Hybrid ML + physics model of the Earth's atmosphere
Coordinate Axes for JAX