This python actually builds a graph under the hood which then gets JIT compiled ... | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

cokernel_hacker on May 3, 2024 | parent | context | favorite | on: TORAX is a differentiable tokamak core transport s...

This python actually builds a graph under the hood which then gets JIT compiled for CPU/GPU/TPU.

senseiV on May 3, 2024 [–]

Does a TPU have XLA-graph for GPUs Cuda-graphs? Not sure on TPU theory

dekhn on May 4, 2024 | [–]

TPUs lower XLA into their own instruction set. https://pytorch.org/xla/release/2.3/index.html

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact