Optimizing frontier model training on TPU v7x (Ironwood)