KPPax: preprocess differentiable kinetic ODE for machine learning in atmospheric chemistry
Atmospheric chemical mechanisms are commonly distributed in the language of the Kinetic PreProcessor (KPP), which generates efficient Fortran or C code but does not fit into the differentiable programming frameworks in which machine-learning models are developed. We present KPPax, a kinetic preprocessor that generates differentiable, just-in-time compiled, and vectorisable chemistry code in JAX…
We haven't written up this one. arXiv cs.AI has the full story — the link below goes straight to it.