Michael Benayoun e62ee803f6 FSDP x TP Saving / Loading, Optimizer (#49256) ...

* Revert "refactor: restore optimizer changes for stacking"

This reverts commit e4b7ba0e9b.

* fix: update guards as they now return booleans

* feat: use reverse logic in load_optimizer_distributed

* refactor: rename helper function

* style: ruff

2026-10-15 15:09:11 +00:00

..

test_checkpointing.py

FSDP x TP Saving / Loading, Optimizer (#49256)

2026-10-15 15:09:11 +00:00