Enabling PyTorch on XLA Devices (e.g. Google TPU)
-
Updated
Sep 21, 2024 - C++
Enabling PyTorch on XLA Devices (e.g. Google TPU)
XLA integration of Open Neural Network Exchange (ONNX)
JAX Foreign Function Interface experiments
Scala (XLA) Stratum protocol proxy with panthera (not official)
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."