Ray
serve build
Gets the correct torch device to use for training.
PublicAPI (beta): This API is in beta and may change before becoming stable.