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


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.

Submitted to the Journal of Open Scientific Software
Raghav Kansal
Raghav Kansal
PhD Candidate in Particle Physics and Machine Learning