Regression for constituents

At the moment we can regress jet-level quantities, but we can't regress all the tracks in the jet. This might be useful as an "aux task" for the jet-level regression