xla

eecf89a2 - [PJRT] Flatten output of `pjrt.run_multiprocess` (#3996)

[PJRT] Flatten output of `pjrt.run_multiprocess` (#3996) * Flatten output of `pjrt.run_multiprocess`. * Update tests * spawn support * Call from `xla_multiprocess`