**Is your feature request related to a problem? Please describe.** pytorch 1.9.1 is available, would be great to include it in the testing pipelines