Comment by cokernel_hacker

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