xla

3bdb4b10 - Fix FSDP for Models with Frozen Weights (#5484)

Fix FSDP for Models with Frozen Weights (#5484) * Fix fsdp not freeing forzen full params * add test * formatting * remove unnecessary env var in test