What metrics would you use to measure an AI agent's performance when optimizing a PyTorch training pipeline for multi-GPU setups?
