PWACG: Partial Wave Analysis Code Generator supporting Newton-conjugate gradient method

Abstract: This paper introduces a novel Partial Wave Analysis Code Generator (PWACG) that automatically generates high-performance partial wave analysis codes. This is achieved by leveraging the JAX automatic differentiation library and the jinja2 template engine. The resulting code is constructed using the high-performance API of JAX, and includes support for the Newton's Conjugate Gradient optimization method, as well as the full utilization of parallel computing capabilities offered by GPUs. By harnessing these advanced computing techniques, PWACG demonstrates a significant advantage in efficiently identifying global optimal points compared to conventional partial wave analysis software packages.