These are some suggestions for updating the README. Maybe you will not be ok with all of them, I deleted relatively much.
In terms of the requirements, I think it's redundant to have some of the requirements listed in the README. Also it is strange to list some but not all of the requirements. The requirements are installed automatically by pip anyways, and also the version pinning is not necessary (e.g., it also works fine with pythron 3.11, pytorch 2+, hydra 1.3 etc).
These are some suggestions for updating the README. Maybe you will not be ok with all of them, I deleted relatively much.
In terms of the requirements, I think it's redundant to have some of the requirements listed in the README. Also it is strange to list some but not all of the requirements. The requirements are installed automatically by
pipanyways, and also the version pinning is not necessary (e.g., it also works fine with pythron 3.11, pytorch 2+, hydra 1.3 etc).