Batched reward model inference and Best-of-N sampling
raw.sh