Google usually talks about it's hardware few months after they have moved to a newer better version. They probably are already using TPUv4 by now.
(disclaimer: while I'm part-time at Google, this is my personal impression, not an official statement, etc., etc.)
(420 TFlop/s, 128GB HBM)