JetNet: A Python package for accessing open datasets and benchmarking machine learning methods in high energy physics

Abstract

JetNet is a Python package that aims to increase accessibility and reproducibility for machine learning (ML) research in high energy physics (HEP), primarily related to particle jets. Based on the popular PyTorch ML framework, it provides easy-to-access and standardized interfaces for multiple heterogeneous HEP datasets and implementations of evaluation metrics, loss functions, and more general utilities relevant to HEP.

Raghav Kansal

PhD Candidate in Particle Physics and Machine Learning