pytorch/xla

Health v2 · maintenance only

Flags: no_license

Adoption not part of the score

2803 stars · 570 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

PyTorch/XLA is a Python package that connects the PyTorch deep learning framework to XLA devices such as Google Cloud TPUs via the XLA deep learning compiler. It enables training and inference of PyTorch models on TPU hardware, with distributed support via PJRT and custom kernel support through Pallas.

Use cases

• train pytorch models on google cloud tpu
• run stable diffusion on tpu
• distributed pytorch training on tpu vms
• port existing pytorch code to tpu hardware
• run pytorch on kaggle free tpu
• write custom tpu kernels with pallas

When to choose

• you need to run PyTorch workloads on Google Cloud TPUs or Kaggle TPU VMs
• you want distributed PyTorch training across TPU pods with PJRT
• you want to experiment with PyTorch on TPU for free via Kaggle notebooks

When to avoid

• you only train on GPUs or CPUs - standard PyTorch suffices
• you need a fully native TPU backend - wait for TorchTPU, which will replace this project
• you depend on pre-C++11 ABI wheels or Python versions older than 3.11 in recent releases

Facets

library · maturity active

deep-learning compiler machine-learning gpu-computing deep-learning machine-learning gpu-computing python cloud pytorch tpu xla torchtpu lazy-tensor-tracing distributed-training linux

Member repositories

For agents

markdown · JSON · MCP: product_card(name="pytorch/xla")