feat(build): Update alpa's build of XLA and PyTorch

[jon-chuang]

trafficstars

System information

• Are you willing to contribute it (Yes/No): Maybe

Describe the new feature and the current behavior/state Currently, we vendor modified tensorflow-alpa and use functorch repo. But functorch is now merged into pytorch main and XLA is now split out into openxla.

Will this change the current API? How? No.

Describe alternatives you've considered

Additional context