I'm wondering if it's possible to wrap a model with fully_shard and use ignored_params to enable AMP for a TP-only or PP-only configuration, without actually sharding the parameters. The goal is to use FSDP just for its parameter management features. I also want to know if the FSDP wrapper itself would cause any performance degradation in this case.