29 November 2021 to 3 December 2021
Virtual and IBS Science Culture Center, Daejeon, South Korea
Asia/Seoul timezone

MadJax - differentiable Matrix Elements with JAX

contribution ID 768
30 Nov 2021, 17:00
20m
S305 (Virtual and IBS Science Culture Center)

S305

Virtual and IBS Science Culture Center

55 EXPO-ro Yuseong-gu Daejeon, South Korea email: library@ibs.re.kr +82 42 878 8299
Oral Track 3: Computations in Theoretical Physics: Techniques and Methods Track 3: Computations in Theoretical Physics: Techniques and Methods

Speakers

Lukas Alexander Heinrich (CERN) Michael Aaron Kagan (SLAC National Accelerator Laboratory (US))

Description

We introduce the differentiable simulator MadJax, an implementation of the general purpose matrix element generator Madgraph integrated within the Jax differentiable programming framework in Python. Integration is performed during automated matrix element code generation and subsequently enables automatic differentiation through leading order matrix element calculations. Madjax thus facilitates the incorporation of high energy physics domain knowledge, encapsulated in simulation software, into gradient based learning and optimization pipelines. In this paper we present the MadJax framework as well as several example applications enabled uniquely through the capabilities of differentiable simulation.

Speaker time zone Compatible with Europe

Primary authors

Lukas Alexander Heinrich (CERN) Michael Aaron Kagan (SLAC National Accelerator Laboratory (US))

Presentation materials