Yes that should work, only thing is you need to call Task init on the master process (and make sure you call Task.current_task() on the subprocesses, if you want to automagic to kick in, that said, usually there is no need, they are supposed to report everything back to the main one anyhow
basically
@call_parse def main(    gpus:Param("The GPUs to use for distributed training", str)='all',    script:Param("Script to run", str, opt=False)='',    args:Param("Args to pass to script", nargs='...', opt=False)='' ):    "PyTorch distributed training launch helper that spawns multiple distributed processes" task = Task.init(...)    current_env = os.environ.copy()