KPPax Generates Differentiable JAX Chemistry Code from KPP Mechanisms
October 28, 2026
KPPax converts Kinetic PreProcessor (KPP) atmospheric chemistry mechanisms into differentiable, JIT-compiled, and vectorisable JAX code. It utilizes sparse operations in log space and integrates with the diffrax library to provide exact sensitivities via forward- and reverse-mode adjoints.
HOW THIS AFFECTS YOU
●
builderThis enables seamless integration of legacy KPP mechanisms into modern machine learning workflows.
●
researcherYou can now perform gradient-based optimization on stiff atmospheric chemical systems using JAX.