An array-oriented Python interface for FastJet

Aryan Roy, Jim Pivarski, Chad Wells Freer

Research output: Contribution to journalConference articlepeer-review

Abstract

Analysis on HEP data is an iterative process in which the results of one step often inform the next. In an exploratory analysis, it is common to perform one computation on a collection of events, then view the results (often with histograms) to decide what to try next. Awkward Array is a Scikit-HEP Python package that enables data analysis with array-at-a-time operations to implement cuts as slices, combinatorics as composable functions, etc. However, most C++ HEP libraries, such as FastJet, have an imperative, one-particle-at-a-time interface, which would be inefficient in Python and goes against the grain of the array-at-a-time logic of scientific Python. Therefore, we developed fastjet, a pip-installable Python package that provides FastJet C++ binaries, the classic (particle-at-a-time) Python interface, and the new array-oriented interface for use with Awkward Array. The new interface streamlines interoperability with scientific Python software beyond HEP, such as machine learning. In one case, adopting this library along with other array-oriented tools accelerated HEP analysis code by a factor of 20. It was designed to be easily integrated with libraries in the Scikit-HEP ecosystem, including Uproot (file I/O), hist (histogramming), Vector (Lorentz vectors), and Coffea (high-level glue). We discuss the design of the fastjet Python library, integrating the classic interface with the array oriented interface and with the Vector library for Lorentz vector operations. The new interface was developed as open source.

Original languageEnglish (US)
Article number012011
JournalJournal of Physics: Conference Series
Volume2438
Issue number1
DOIs
StatePublished - 2023
Event20th International Workshop on Advanced Computing and Analysis Techniques in Physics Research, ACAT 2021 - Daejeon, Virtual, Korea, Republic of
Duration: Nov 29 2021Dec 3 2021

All Science Journal Classification (ASJC) codes

  • General Physics and Astronomy

Fingerprint

Dive into the research topics of 'An array-oriented Python interface for FastJet'. Together they form a unique fingerprint.

Cite this