Konečně jsem se dostal do bodu, kdy dost věcí, na které spoléhám, je b200 + cuda + hořák specifický, takže jsem nemohl dosáhnout stejných výsledků na TPU s Jaxem. Jaký nečekaný zvrat událostí.