Add torch to the build