Missing documentation on how to use the launcher with GPU programs such that each job runs on a different GPU?

[mauriciogtec]

Is there any documentation/examples on how to use the launcher with a GPU Tensorflow or Pytorch program correctly? Specifically, is it possible to use the launcher such that each job is assigned a different GPU of a multi-GPU node? Something like --gpus-per-task 1.