论文标题
Madjax可区分的矩阵元素
Differentiable Matrix Elements with MadJax
论文作者
论文摘要
Madjax是生成和评估高能散射过程的可区分矩阵元素的工具。因此，这是迈向高能物理学的可区分编程范式的一步，该范式促进了在模拟软件中编码的高能量物理领域知识的融合到基于梯度的学习和优化管道中。 Madjax包括两个组成部分：（a）通用矩阵元素生成器madgraph的插件，该插件将矩阵元素和相位空间采样代码与JAX可区分的编程框架集成在一起，以及（b）独立包装API，用于访问与自动差异计算的矩阵元素及其梯度。介绍了基于模拟的推理和基于流量的矩阵元素建模的MADJAX实现和示例应用程序，并介绍了具有可区分矩阵元素的功能。
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 API 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.
