11–15 Mar 2024
Charles B. Wang Center, Stony Brook University
US/Eastern timezone

dilax: Differentiable Binned Likelihoods in JAX

14 Mar 2024, 17:50
20m
Lecture Hall 2 ( Charles B. Wang Center, Stony Brook University )

Lecture Hall 2

Charles B. Wang Center, Stony Brook University

100 Circle Rd, Stony Brook, NY 11794
Oral Track 2: Data Analysis - Algorithms and Tools Track 2: Data Analysis - Algorithms and Tools

Speaker

Manfred Peter Fackeldey (RWTH Aachen University (DE))

Description

dilax is a software package for statistical inference using likelihood
functions of binned data. It fulfils three key concepts: performance,
differentiability, and object-oriented statistical model building.
dilax is build on JAX - a powerful autodifferentiation Python frame-
work. By making every component in dilax a “PyTree”, each compo-
nent can be jit-compiled (jax.jit), vectorized (jax.vmap) and differ-
entiated (jax.grad). This enables additionally novel computational
concepts, such as running thousands of fits simultaneously on a GPU
or differentiating through measurements of physical observables.
We present the key concepts of dilax, show its features, and discuss
performance benchmarks with toy datasets.

Significance

This project is a new statistics tool suited for typical measurements in LHC analyses. It focusses on performance, usability, and novel computing techniques such as autodifferentiation and vectorization of full fits. This project has not been presented so far. It introduces new concepts that have not been covered by other statistics libraries.

Experiment context, if any Use case for CMS, ATLAS, LHCb analyses

Primary authors

Manfred Peter Fackeldey (RWTH Aachen University (DE)) Benjamin Fischer (RWTH Aachen University (DE)) Felix Philipp Zinn (Rheinisch Westfaelische Tech. Hoch. (DE)) Martin Erdmann (Rheinisch Westfaelische Tech. Hoch. (DE))

Presentation materials