about extra
  • [Sep 7, 2026] Async RL in Pure JAX
  • [Jul 6, 2026] MIRA: Multiplayer Interactive World Models with Representation Autoencoders
    ▸ Demo
  • [Dec 13, 2025] Composite Classifier-Free Guidance for Multi-Modal Conditioning in Wind Dynamics Super-Resolution
  • [Sep 6, 2025] Jaxformer: Scaling MoE pretraining on TPUs
    ▸ Talk @ Google JAX Developer Conference