[functorch] Added docstring for `grad` (pytorch/functorch#231)
* Added docstring for `grad`
Description:
- Addresses pytorch/functorch#224
- WIP on adding docs for grad
- Cleaned unused imports and other nits
* [skip ci] Added note on torch.no_grad and ft grad usage
* Updated docstring
