Allow individual layouts in the device_layouts
argument of `PjRtCli…
#13254
This job was skipped
Loading
device_layouts
argument of `PjRtCli…
#13254