xla

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

GitHub

2k stars
57 watching
467 forks
Language: C++
last commit: 8 days ago
compilerdeep-learningpytorchxla

Screenshot of pytorch/xla website

pytorch.org/xla