Saya akhirnya sampai pada titik di mana cukup banyak hal yang saya andalkan adalah b200 + cuda + obor khusus sehingga saya tidak bisa mendapatkan hasil yang sama pada TPU dengan Jax. Sungguh pergantian peristiwa yang tidak terduga.