TPUs (via XLA integrations)

E96636

PyTorch accelerator backend XLA-based compilation target hardware accelerator integration

TPUs (via XLA integrations) are Google's specialized tensor processing units that can be used as accelerators for PyTorch models through the XLA compilation framework.

Try in SPARQL Jump to: Surface forms Disambiguation Statements Elicitation Referenced by

AI illustration

All labels observed (6)

Label	Occurrences
PyTorch (via XLA / integration layers)	1
PyTorch (via XLA integration)	1
PyTorch/XLA runtime	1
TPUs	1
TPUs (via XLA integrations) canonical	1
XLA (Accelerated Linear Algebra)	1

How this entity was disambiguated

This entity first appeared as the object of triple T825523 — resolving that mention is where its identity was fixed. The disambiguator weighed these candidate entities and picked the highlighted one (or “None”, minting a new entity). This is how homonymy is resolved: the same surface form can point to different entities.

NED1 Entity disambiguation (via context triple) gpt-5-mini-2025-08-07

Target entity: TPUs (via XLA integrations)
Context triple: [PyTorch, supportsHardware, TPUs (via XLA integrations)]

A. TensorFlow
TensorFlow is an open-source, end-to-end machine learning and deep learning framework widely used for building, training, and deploying neural network models at scale.
B. Google Tensor
Google Tensor is Google's custom-designed system-on-a-chip (SoC) platform created to power Pixel devices with advanced AI and machine learning capabilities.
C. NVIDIA CUDA
NVIDIA CUDA is a parallel computing platform and programming model that enables developers to use NVIDIA GPUs for general-purpose high-performance computing.
D. RLlib
RLlib is a scalable, open-source reinforcement learning library built on Ray that provides high-level APIs and distributed training support for a wide range of RL algorithms.
E. PyTorch
PyTorch is an open-source deep learning framework widely used for building and training neural networks, known for its dynamic computation graph and strong support for research and production in Python.
F. None of above. chosen
G. Unsure - the case is ambiguous/there is not enough information to decide.

NED2 Entity disambiguation (via description) gpt-5-mini-2025-08-07

Target entity: TPUs (via XLA integrations)
Target entity description: TPUs (via XLA integrations) are Google's specialized tensor processing units that can be used as accelerators for PyTorch models through the XLA compilation framework.

A. TensorFlow
TensorFlow is an open-source, end-to-end machine learning and deep learning framework widely used for building, training, and deploying neural network models at scale.
B. Google Tensor
Google Tensor is Google's custom-designed system-on-a-chip (SoC) platform created to power Pixel devices with advanced AI and machine learning capabilities.
C. NVIDIA CUDA
NVIDIA CUDA is a parallel computing platform and programming model that enables developers to use NVIDIA GPUs for general-purpose high-performance computing.
D. RLlib
RLlib is a scalable, open-source reinforcement learning library built on Ray that provides high-level APIs and distributed training support for a wide range of RL algorithms.
E. PyTorch
PyTorch is an open-source deep learning framework widely used for building and training neural networks, known for its dynamic computation graph and strong support for research and production in Python.
F. None of above. chosen

Statements (47)

Predicate	Object
instanceOf	PyTorch accelerator backend ⓘ XLA-based compilation target ⓘ hardware accelerator integration ⓘ
abstracts	low-level TPU device management ⓘ
aimsTo	accelerate deep learning workloads ⓘ reduce training time for large models ⓘ
benefits	users needing scalable training on Google Cloud TPUs ⓘ
category	hardware-accelerated deep learning backend ⓘ machine learning infrastructure ⓘ
compatibleWith	Google Cloud TPU V2 ⓘ Google Cloud TPU V3 ⓘ Google Cloud TPU V4 ⓘ
designedFor	high-throughput tensor operations ⓘ large batch training ⓘ
developedBy	Google ⓘ
documentationHostedAt	https://github.com/pytorch/xla ⓘ
enables	accelerated tensor computations ⓘ execution of PyTorch models on TPUs ⓘ graph compilation via XLA ⓘ
exposes	XLA-specific debugging tools ⓘ profiling utilities for TPU workloads ⓘ
handles	automatic differentiation on TPU via XLA graphs ⓘ
integratesWith	PyTorch autograd system via XLA ⓘ
mapsTo	TPU cores as PyTorch devices ⓘ
optimizationMethod	ahead-of-time compilation ⓘ graph-level optimization ⓘ operation fusion ⓘ
partOf	PyTorch/XLA project ecosystem ⓘ linked to: XLA
provides	PyTorch-like APIs for TPU execution ⓘ device placement utilities ⓘ distributed data loader support ⓘ
requires	PyTorch/XLA runtime ⓘ linked to: TPUs (via XLA integrations) XLA compiler ⓘ XLA-compatible PyTorch operations ⓘ specialized input pipelines for TPUs ⓘ
supports	data parallel training ⓘ distributed training ⓘ mixed precision training ⓘ model parallel training ⓘ synchronous data parallelism across TPU cores ⓘ
supportsFramework	PyTorch ⓘ PyTorch/XLA ⓘ linked to: PyTorch
targetHardware	Google TPU ⓘ linked to: Tensor Processing Unit
usedFor	inference of deep learning models ⓘ training neural networks ⓘ
usedIn	large-scale machine learning experiments ⓘ
usesFramework	XLA ⓘ

How these facts were elicited