Hi, was wondering if there's any way to support multi-gpu at this point other than setting up the env var CUDA_VISIBLE_DEVICES thank you