Differentiable Matrix Elements with MadJax

02/28/2022
by   Lukas Heinrich, et al.
0

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.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
09/20/2020

Topology Optimization through Differentiable Finite Element Solver

In this paper, a topology optimization framework utilizing automatic dif...
research
05/03/2022

Differentiable Simulation of Soft Multi-body Systems

We present a method for differentiable simulation of soft articulated bo...
research
10/08/2021

Differentiable Programming of Isometric Tensor Networks

Differentiable programming is a new programming paradigm which enables l...
research
12/07/2020

Using Differentiable Programming for Flexible Statistical Modeling

Differentiable programming has recently received much interest as a para...
research
11/29/2021

A new Sinkhorn algorithm with Deletion and Insertion operations

This technical report is devoted to the continuous estimation of an epsi...
research
08/31/2023

Branches of a Tree: Taking Derivatives of Programs with Discrete and Branching Randomness in High Energy Physics

We propose to apply several gradient estimation techniques to enable the...
research
03/15/2022

New directions for surrogate models and differentiable programming for High Energy Physics detector simulation

The computational cost for high energy physics detector simulation in fu...

Please sign up or login with your details

Forgot password? Click here to reset