Move torch config to device specific config file