Skip to content

ONNX inference for BoostedJetTaggers tool (only ONNX utils)

Antonio Giannini requested to merge angianni/athena:angianni-bjt-rel25-ONNXxBJT into main

This MR aims to add the functionalities to the BJT package to retrive ML models for jet tagging group. Other MRs will follow to 1. turn on the configuration at the derivation level and 2. add WPs and SFs for new taggers.

The main design is based on:

  • JSSMLTool: utils to run the ONNX inference
  • JSSTaggerUtils: utils to load/prepare all the jet level inputs

This follows/improve the one done in rel.21 already with adding more functionalities for new taggers and optimisation of the code: !50965 (merged)

It has also been discussed within the JetTagging group: https://indico.cern.ch/event/1431633/contributions/6023131/attachments/2886096/5058425/JetTagging_meeting_27jun24.pdf

Thanks in advance!

Cheers,

Antonio

Tagging:

@cdelitzs , @fballi , @jdandoy

@rles , @jzahredd , @stapiaar , @bhodkins

Edited by Antonio Giannini

Merge request reports