Tianjun Zhang(@tianjun_zhang):We scale up RL on TPUs for the MiMo families 🚀 It is quite a journey for us to bring RL to scale with Jax + TPU: the most amazing part for me is that scaling it up is mostly a config change, not a code rewrite. This is the dream for any researcher 🫣 Read our blog for more details. Thanks to @XiaomiMiMo, @berkeley_ai, and @googlecloud for making it happen! We'll share more details soon!

Tianjun Zhang

@tianjun_zhang

@peano_ai | Prev @meta @GoogleDeepMind @berkeley_ai | Project Lead of LiveCodeBench, RAFT and Gorilla LLM

参加 March 2017

1.2K フォロー中 1.7K ファン

Tianjun Zhang@tianjun_zhang

2026.09.21 21:47

We scale up RL on TPUs for the MiMo families 🚀 It is quite a journey for us to bring RL to scale with Jax + TPU: the most amazing part for me is that scaling it up is mostly a config change, not a code rewrite. This is the dream for any researcher 🫣 Read our blog for more details. Thanks to @XiaomiMiMo, @berkeley_ai, and @googlecloud for making it happen! We'll share more details soon!

127

コミュニティへ転送

人気のあるユーザー