pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
2,489Updated this week

Related projects

Alternatives and complementary repositories for xla