xla

89574d2d - Fix FSDP for Models with Frozen Weights (#5484) (#5539)

Fix FSDP for Models with Frozen Weights (#5484) (#5539) * Fix fsdp not freeing forzen full params * add test * formatting * remove unnecessary env var in test Co-authored-by: Liyang90 <liyanglu@google.com>