ついに、頼っているものの多くがb200 + cuda + torch専用で、JaxのTPUでは同じ結果が得られないレベルに達しました。なんて予想外の展開でしょう。