Supports multi-GPU processing with torchrun for faster computation.