Remove torch

Change-Id: Ic6d013d657d63d19389bcefbab6e7ce5ca254540