torchrun --nproc-per-node N script.py