We had to remove Python 3.10 from the CI when migrating to this repo. We should update package to support Python 3.10 and revert https://github.com/pytorch/MultiPy/commit/3cedbb8c63e657800ac6a2bce06578a9cf874af4
