xla
[FSDP] Don't move full tensor to device in deferred_init
#4819
Merged

Loading