Comment on TORAX is a differentiable tokamak core transport simulatorparentComments−cokernel_hacker2yThis python actually builds a graph under the hood which then gets JIT compiled for CPU/GPU/TPU.−senseiV2yDoes a TPU have XLA-graph for GPUs Cuda-graphs? Not sure on TPU theory−dekhn2yTPUs lower XLA into their own instruction set. https://pytorch.org/xla/release/2.3/index.html
Comments
This python actually builds a graph under the hood which then gets JIT compiled for CPU/GPU/TPU.
Does a TPU have XLA-graph for GPUs Cuda-graphs? Not sure on TPU theory
TPUs lower XLA into their own instruction set. https://pytorch.org/xla/release/2.3/index.html