fix: return gradients on original device for multi-GPU training#423

Vinayyyy7 wants to merge 1 commit intounslothai:mainunslothai/unsloth-zoo:mainfrom

Vinayyyy7:fix/multi-gpu-gradient-device-mismatchVinayyyy7/unsloth-zoo:fix/multi-gpu-gradient-device-mismatchCopy head branch name to clipboard