**Is your feature request related to a problem? Please describe.** - add tests for the new pytorch releases - drop the tests for v1.7.x?