fastmachinelearning / SonicCMS

Services for Optimized Network Inference on Coprocessors (for CMS)
8 stars 8 forks source link

Add HGCal graph NN producers #16

Open kpedro88 opened 4 years ago

kpedro88 commented 4 years ago

The HGCal graph NN-based reconstruction has not been officially introduced in CMSSW. Implementation in a SonicTriton producer is likely the easiest path (especially for a PyTorch graph-based network, since the PyTorch C++ API is not yet available in CMSSW). There are multiple networks to implement, including edge classification (for clustering) and dynamic reduction (for energy regression etc.).

Particular considerations:

Assigned to: @tklijnsma, @lgray

kpedro88 commented 3 years ago

Relevant: https://github.com/lgray/triton-torchgeo-gat-example