[PJRT:GPU] Don't stage transfers greater than 1GB.
[PJRT:GPU] Don't stage transfers greater than 1GB.

Allocating pinned memory of that size takes a significant amount of time on GPU.

Fixes https://github.com/google/jax/issues/20209
