Speaker
Manfred Peter Fackeldey
(RWTH Aachen University (DE))
Description
dilax
is a software package for statistical inference with binned likelihoods. It focusses on three key concepts: performance, differentiability, and object-oriented statistical model building. Thus, dilax
is build upon the shoulders of a deep learning giant: JAX - a popular autodifferentiation Python framework. By making every component in dilax
a PyTree
, each component can be jit-compiled (jax.jit
), vectorized (jax.vmap
) and differentiated (jax.grad
). This does not only fulfil all key concepts, but also enables novel computational concepts, such as running thousands of fits simultaneously on a GPU.
We present the key concepts of dilax
, show its features, and discuss performance benchmarks with toy datasets.
Author
Manfred Peter Fackeldey
(RWTH Aachen University (DE))