wandb run: https://wandb.ai/eleutherai/pythia-rlhf/runs/hlfywf2d