Docs setup needs to be upgraded to use PyTorch Sphinx Theme2

[divyanshk]

🐛 Describe the bug

More info: https://github.com/pytorch/pytorch_sphinx_theme/blob/pytorch_sphinx_theme2/pytorch_sphinx_theme2_migration_guide.md

Versions

.