Masaki Kozuki
Masaki Kozuki
Collect popular models' `GraphModule`s with distributed collective communication operators in there
## What does this PR do? because it should be installed when thunder is installed.
## 🐛 Bug `test_networks.py::test_checkpointing_thunderfx` fails due to grads mismatch between eager pytorch and thunderfx. ### To Reproduce Steps to reproduce the behavior: 1. Run `test_networks.py::test_checkpointing_thunderfx` 1. See grad mismatch e.g....
as the linked issue is fixed https://github.com/Lightning-AI/lightning-thunder/blob/bf1fcf2f66c04642cb31517762bed367b093296c/thunder/tests/test_update_aliases.py#L217-L226 cc @tfogal