وصلت أخيرا إلى مرحلة أن الكثير من الأشياء التي أعتمد عليها هي b200 + cuda + torch الخاصة ب Shoulder، لذلك لم أتمكن من الحصول على نفس النتائج على وحدات TPU مع Jax. يا له من تحول غير متوقع في الأحداث.