Make dependency on Torch explicit