Differentiable Matrix Elements with MadJax. Issue 1 (1st February 2023)

• Record Type: Journal Article
• Authors: Heinrich, Lukas; Kagan, Michael
• Is Part Of: Journal of physics. Volume 2438:Issue 1(2023)
• Publication Date: 2023-02-01
• Subjects: Physics -- Congresses; 530.5
• DOI: 10.1088/1742-6596/2438/1/012137
• Languages: English
• ISSNs: 1742-6588

Abstract

Abstract: MadJax is a tool for generating and evaluating differentiable matrix elements of high energy scattering processes. As such, it is a step towards a differentiable programming paradigm in high energy physics that facilitates the incorporation of high energy physics domain knowledge, encoded in simulation software, into gradient based learning and optimization pipelines. MadJax comprises two components: (a) a plugin to the general purpose matrix element generator MadGraph that integrates matrix element and phase space sampling code with the JAX differentiable programming framework, and (b) a standalone wrapping code interface for accessing the matrix element code and its gradients, which are computed with automatic differentiation. The MadJax implementation and example applications of simulation based inference and normalizing flow based matrix element modeling, with capabilities enabled uniquely with differentiable matrix elements, are presented.