Implement `PjRtStreamExecutorDevice::default_memory_space()` for cases where there's only one memory space
Implement `PjRtStreamExecutorDevice::default_memory_space()` for cases where there's only one memory space
