Add unit tests for ClippedPGLossFn (especially for upcoming vocab parallel path).

Current unit tests only cover the NLL loss. Should also have tests for ClippedPGLossFn (superset of REINFORCE, PPO, GRPO, DAPO losses). Upcoming PR adds vocab parallel to the loss function as well