first draft of GPU training with SLURM