Add sharding devices to XlaCompileOptions and plumb them through from JAX.

[copybara-service[bot]]

Add sharding devices to XlaCompileOptions and plumb them through from JAX.

This is necessary to support MPMD parallelism in McJAX, since the PjRt-IFRT executable's output shardings can no longer be built with the addressable devices from the PJRT executable, in the case where the executable has no addressable devices.