Fast Jet Finding in Julia
Graeme Andrew Stewart. Sanmay Ganguly, Sattwamo Ghosh, Philippe Gras, Atell Krasnopolski
TL;DR
The paper addresses the need for fast, flexible jet reconstruction in collider data analysis by introducing JetReconstruction.jl, a high-performance Julia package for sequential jet finding. It details a production-ready release with support for both $pp$ and $e^+e^-$ collision algorithms, including Anti-$k_T$, Cambridge/Aachen, Inclusive-$k_T$, Durham, and Generalised $k_T$, along with SIMD-enhanced core routines and a ClusterSequence data model. Key contributions include the introduction of an $(\theta,\phi)$-space EEjet for $e^+e^-$ clustering, EDM4hep data model support, and substructure tools (Soft Drop, Mass Drop, Filtering, Trimming), backed by extensive benchmarking showing speedups over FastJet. The work enables fast, scalable jet reconstruction in Julia, with practical impact for FCCee and future colliders, and provides a pathway for broader analysis tooling and data-format interoperability within the Julia ecosystem.
Abstract
Jet reconstruction remains a critical task in the analysis of data from HEP colliders. We describe in this paper a new, highly performant, Julia package for jet reconstruction, JetReconstruction.jl, which integrates into the growing ecosystem of Julia packages for HEP. With this package users can run sequential reconstruction algorithms for jets. In particular, for LHC events, the Anti-${k}_\text{T}$, Cambridge/Aachen and Inclusive-${k}_\text{T}$ algorithms can be used. For FCCee studies the use of alternative algorithms such as the Generalised ${k}_\text{T}$ for $e^+e^-$ and Durham are also supported. The performance of the core algorithms is better than Fastjet's C++ implementation, for typical LHC and FCCee events, thanks to the Julia compiler's exploitation of single-instruction-multiple-data (SIMD), as well as ergonomic compact data layouts. The full reconstruction history is made available, allowing inclusive and exclusive jets to be retrieved. The package also provides the means to visualise the reconstruction. Substructure algorithms have been added that allow advanced analysis techniques to be employed. The package can read event data from EDM4hep files and reconstruct jets from these directly, opening the door to FCCee and other future collider studies in Julia.
