How to use xla_fsdp?

Fabric
1

could you give an example how xla_fsdp works on tpu? I couldn’t get it work. loss does not decrease