xla

c4daf6ab - Backport Enable eager spmd #7341 and fix eager mode spmd module loading with fsdpv2 #7631 (#7673)