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))