This also adds Python 3.9 compatibility.
Changelog: https://github.com/pytorch/pytorch/releases/tag/v1.7.0