Dan Saunders ae1c7ace63 Sequence parallel training context manager (#2553) ...

* ctx manager for SP

* updates

* update

* further simplifying

* accommodate both training context managers

* simplifying

* simplifying

* nit

* reorg

* tweak codecov yaml

* add gather post hook, simplify, fixes

* pytest

* pytest fix

2025-04-25 10:33:54 -04:00

3.5 KiB

Raw Permalink Blame History

View Raw