Pyrometheus: Symbolic abstractions for XPU and automatically differentiated computation of combustion kinetics and thermodynamics

The cost of combustion simulations is often dominated by the evaluation of net production rates of chemical species and mixture thermodynamics (thermochemistry). Execution on computing accelerators (XPUs) such as graphics processing units (GPUs) can greatly reduce this cost. Established thermochemistry software is not readily portable to such devices, as it sacrifices valuable analytical forms that enable differentiation, sensitivity analysis, and implicit time integration. Symbolic abstractions are developed with corresponding transformations that enable computation on accelerators and automatic differentiation by avoiding premature specification of detail. The software package Pyrometheus is introduced as an implementation of these abstractions and their transformations for combustion thermochemistry. The formulation facilitates code generation from the symbolic representation of a specific thermochemical mechanism in multiple target languages, including Python, C ++ , and Fortran. The generated code processes array-valued expressions, but does not specify their semantics. The semantics are provided by compatible array libraries, including NumPy, Pytato, and Google JAX. Thus, the generated code retains a symbolic representation of the thermochemistry, which enables computation on accelerators and CPUs and facilitates automatic differentiation. The design and operation of the symbolic abstractions and their companion tool, Pyrometheus, are discussed throughout. Roofline demonstrations show that the computation of chemical source terms within MFC, a Fortran-based flow solver we link to Pyrometheus, is performant.