revert accidentally deleted gpus in example
revert accidentally deleted gpus in example

PiperOrigin-RevId: 373251842
Change-Id: I99ef3598d1d5f8727fe2655c4dcc62ffd780cbe8
GitOrigin-RevId: 1a6c6f0e2cc353dfd8621c56261b23c7334ed711

deepmind/xmanager,deepmind/xmanager