Skip to content

[TPU] Add support for PJRT from PyTorch/XLA 2.0#17352

Merged
carmocca merged 56 commits intoLightning-AI:masterfrom Liyang90:xla_pjrt_debugApr 18, 2023

Commits

Commits on Mar 20, 2023

Commits on Mar 21, 2023

Commits on Apr 11, 2023

Commits on Apr 12, 2023

Commits on Apr 13, 2023

Commits on Apr 14, 2023

Commits on Apr 15, 2023

Commits on Apr 16, 2023

Commits on Apr 17, 2023

Commits on Apr 18, 2023