Implemented already in heptrkx-gnn-tracking. Add improved support for optimizer choices and LR scaling