Google's new TPU 8i Max uses $ARM based Axion CPUs to prepare data and manage AI workloads, joining TPU 8t and 8i.
That means 3 of Google's 4 next-gen TPU systems now use Arm host CPUs, versus x86 hosts previously.
The fourth, TPU 8tx, uses $AMD EPYC CPUs.