Revert "Revert "remove torch overlay""

This reverts commit 230ebed22a5c95f9f945ea77d831e86505fd7fd3.