27 ms·This python actually builds a graph under the hood which then gets JIT compiled for CPU/GPU/TPU.by cokernel_hacker 2y agoThis python actually builds a graph under the hood which then gets JIT compiled for CPU/GPU/TPU.senseiV 2y agoDoes a TPU have XLA-graph for GPUs Cuda-graphs? Not sure on TPU theorydekhn 2y agoTPUs lower XLA into their own instruction set. https://pytorch.org/xla/release/2.3/index.html https://pytorch.org/xla/release/2.3/index.html