JAX 相关 AI 资讯 · AI 热点

7月26日

03:08

03:08

官方一手marktechpost@Asif Razzaq

Reactor 团队开源了 Open Dreamer，这是一个基于 JAX 和 Flax NNX 的 Dreamer 4 世界模型管道的完整复现。该项目包含两个代码仓库：next-state/open-dreamer 提供训练管道，包括因果视频分词器、动作条件潜在动力学模型、轨迹生成和 FVD 评分；reactor-team/open-dreamer 提供推理代码。Open Dreamer 的发布使 Dreamer 4 的完整训练流程得以公开复现。

AI模型 Open Dreamer Dreamer 4 JAX

推荐理由：Reactor 团队把 Dreamer 4 的完整训练配方和代码都开源了，JAX/Flax 实现，想复现世界模型做视频预测的赶紧看。

7月2日

15:25

15:25

AI Will@FinanceYF5

Aaron Lemke使用Fable 5为其强化学习生物添加了毛发和新的语音。语音基于神经网络的激活状态（左上图显示）。该演示使用了Three.js和WebGL/WebGPU进行渲染。模型在Runpod上使用JAX和MuJoCo（来自Google DeepMind）训练。

AI产品 Fable 5 Three.js JAX

推荐理由：看Aaron Lemke用Fable 5给他的RL生物加上毛发和实时语音，声音还跟着神经网络激活变化，太酷了。

15:20

15:20

AI Will@FinanceYF5

精选

开发者使用Fable 5为强化学习训练出的生物添加了毛发和动态声音。声音并非录制，而是直接从神经网络的激活值实时生成。训练过程基于JAX和MuJoCo模拟器，渲染使用three.js和WebGPU。最终效果看起来像活物在地板上爬行，引发对AI艺术与研究边界的讨论。

AI模型 Fable 5 RL JAX

推荐理由：这个项目把RL训练的活物做得超逼真，毛发和声音都来自神经网络激活值，技术堆叠很酷，值得看一眼。

6月19日

09:55

09:55

官方账号arXiv cs.LG@Tristan Tomilin, Mourad Boustani, Mickey Beurskens, Thiago D. Simão

CRAX基于MuJoCo XLA（MJX）物理引擎，利用JAX的向量化操作和硬件加速，在安全强化学习基准测试中实现约100倍于CPU版本的速度提升。该基准包含六套环境和三类智能体任务，每类任务设三个难度级别。研究者在六个主流安全RL方法上的评估表明，没有一种方法在所有任务中占优，揭示了性能与安全之间的权衡。实验还发现，跨难度级别的课程学习与安全迁移在困难设置下比直接训练效果更好。

AI模型 CRAX MuJoCo JAX

推荐理由：想快速测试安全RL算法？CRAX用JAX把基准跑快100倍，还自带六个环境和三级难度，帮你一眼看出哪个方法在安全与性能上最均衡。

5月13日

00:33

00:33官方一手Google Developers Blog（博客/媒体）

Google的MaxText框架现在支持在单主机TPU上进行监督微调（SFT）和强化学习（RL），利用JAX和Tunix库实现高效模型优化。开发者可通过GRPO和GSPO等算法轻松调整预训练模型以适应专业任务和复杂推理。该更新简化了从单主机到多主机配置的微调流程，为后训练工作流提供了可扩展路径。

AI产品微调/RL TPU JAX

推荐理由：这表明Google正在降低TPU上高级模型微调的门槛，让中小规模团队也能在单主机TPU上应用RL算法进行后训练，对AI开发实践有实用价值。