Add `trace_module` to docs (#24258)
Summary:
Stacked PRs
* **#24258 - [jit] Add `trace_module` to docs**
* #24208 - [jit] Cleanup documentation around `script` and `trace`
](https://our.intern.facebook.com/intern/diff/16811342/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/24258
Pulled By: driazati
Differential Revision: D16811342
fbshipit-source-id: 893be85a711ab180319b790ed1c72b93022373c1