Reference to https://github.com/DL4Jets/DeepJetCore/pull/11
Reference to
#11