Enable Torch

Change-Id: I22e7f7026d63fc6240c208f034e03b335a556c7b