Transfer any JTensors to CPU and enforce running `BaseModel.process_decode_out` on CPU
Transfer any JTensors to CPU and enforce running `BaseModel.process_decode_out` on CPU

PiperOrigin-RevId: 476918305

google/paxml,google/paxml