Since torch is a heavy library, ideally we should figure out how it works and best practices around using it