Lorenz96 Graph Neural Network

I am building a graph neural network (GNN) that predicts the two-sccale Lorenz96 system: a chaotic, toy model of Earth’s climate in my research lab. Combining Python and libraries like JAX, jraph, and scipy, I am creating a smale-scale climate model that examines how well GNNs predict chaotic systems like the atmosphere. I am currently writing the academic paper for this project, so I can’t share code until the paper is published.

Lab Website