Add agent_ref level metrics in training for GRPO and OPD

For current GRPO and OPD runs, the logprob gap / reward variacne / advantage etc seem to be reported and tracked only in aggregate instead of per-environment (from Nemo-Gym). It would be great to add agent_ref level metrics.