Я нарешті дійшов до того рівня, коли достатньо того, на що я покладаюся, стосуються b200 + cuda + пальника, і я не міг отримати однакові результати на TPU з Jax. Який несподіваний поворот подій.