Fix FSDP2 and distributed checkpointing imports for older PyTorch versions (#46141)
* Fix PyTorch requirement for FSDP2 to >=2.6
* Fix PyTorch requirement for distributed checkpoint saving to >=2.7
---------
Co-authored-by: Ferdinand Mom <47445085+3outeille@users.noreply.github.com>