Enable Torch

Change-Id: I69cc406b048451ed4f18f714326ea62189cc2aaf