pytorch/xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)

语言：C++ · GitHub 星标：2,806 · Fork：574

话题：compiler · deep-learning · pytorch · xla

在 GitHub 上查看原仓库