Allow individual layouts in the device_layouts argument of `PjRtCli…
#14710
| Job | Run time |
|---|---|
| 1m 12s | |
| 1m 12s |
device_layouts argument of `PjRtCli…
#14710
| Job | Run time |
|---|---|
| 1m 12s | |
| 1m 12s |