Revert "Track AOKPs torch"

This reverts commit 0a33c693143c3f98212c8bfb115e67e7265cc350.