xla

[FSDP] Replace all .data usages with modern approaches

#4485

Merged