yuga: Move to OmniTorch