Na verdade, não acho que as TPUs tenham uma vantagem significativa sobre as GPUs. A vantagem de usá-las é que 1) o Jax é excelente e 2) elas são gerenciadas e configuradas corretamente porque estão em data centers do Google. Muitos problemas são causados por pessoas que usam clusters de GPUs mal configurados.
Carregando detalhes do thread
Buscando os tweets originais no X para montar uma leitura limpa.
Isso normalmente leva apenas alguns segundos.