When I was there a few years ago, we only got CPUs and GPUs for training. TPUs were in too high of demand.