Allow individual layouts in the device_layouts argument of PjRtClient::CreateBuffersForAsyncHostToDevice to be std::nullopt, meaning that those buffers should use the default device layout.

Allow individual layouts in the device_layouts argument of PjRtClient::CreateBuffersForAsyncHostToDevice to be std::nullopt, meaning that those buffers should use the default device layout. #62351