Revert "Revert "torch to temasek""

This reverts commit 022d65e6728dee7adf0334a9c22563a1f1e22d9d.