Кстати, TPU не золотая пуля. Под него надо писать особый код. Прям очень особый. И даже после заморочек он даст ~30% буста на киловат. Личный опыт работы с Гугл TPU на коллабе.
Нет, нереально. Ты же не можешь написать компилятор питона, эмулирующий циклы без циклов. Ты можешь только запретить прогеру юзать циклы, иначе фиг тебе а не буст на tpu