Use lax.scan in `rewards_to_go` and `deltas`.
Use lax.scan in `rewards_to_go` and `deltas`.

PiperOrigin-RevId: 250584446

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor