Add torch to the build

Change-Id: Ie1218bbb2272ed1ebe3419d3a932a3d8b90a2106