Compatibility with torch version 2.xx and above

This repo is not compatible with pytorch version that is above 2.0, unlike explanation in INSTALL.md
"The _six dependency was removed in 2023" so, "six" in new torch versions are not supported anymore. This repo, however, uses 'six'.

Because H100 supports from sm90, old torch version (1.xx, etc) are not compatible with H100. Thus, this repo can't run in systems that consist of H100.

I would greatly appreciate for updated code so this repo also supports newer torch version.