Remove torch

Change-Id: I5b289ca941b807b2aabd2d893dd0797c746c585c