nvalchemi.distributed.collective_device# nvalchemi.distributed.collective_device(fallback='cpu')[source]# Resolve the rank-local device for distributed tensor collectives. Parameters: fallback (device | str) Return type: device
