Revert "remove torch overlay"

This reverts commit 869c25c3c942eecce6138d6e5e6a2456ce9a6086.