Skip to content

Draft: ONNX inference for BoostedJetTaggers tool

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

This MR aims to add the functionalities to the BJT package to retrive ML models for jet tagging group. Another MR will follow with turning on the configuration at the derivation level.

The main design is based on:

  • JSSMLTool: utils to run the ONNX inference
  • JSSTaggerUtils: utils to load/prepare all the jet level inputs
  • SmootherXXXTagger: the historical tagger tool that performs the WP cut/SF retrieve (the ML score will replace historical variables)

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: <will/do/later>

Merge request reports