This PR:
- dedups our colab notebooks with the regular functorch notebooks. The
colab notebooks were versions of the reuglar notebooks that had
install instructions. Now that functorch is easier to install, we do not
need those anymore.
- fixes the colab links
Test Plan:
- build docs locally and tested them
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85855
Approved by: https://github.com/samdow
Fix a small bug
```python3
if compute == 'full':
return result
if compute == 'trace':
return torch.einsum('NMKK->NM') # should be torch.einsum('NMKK->NM', result)
if compute == 'diagonal':
return torch.einsum('NMKK->NMK') # should be torch.einsum('NMKK->NMK', result)
```
