#JAX

共 10 条 · 7 天 1 条 · 30 天 3 条

Double-Stitch 1 Dreamer 4 1 Flax 1 GPU 1 GRADSOLVE 1 Jaxolotl 1 KerasHub 1 LTL 1

信息流里打上「JAX」标签的资讯、产品与论文，按刊登时间排，新的在上。

10月11日

技巧 Agent 与开发者09:41

KerasHub+JAX+TPU 做 LLM 后训练实验

Keras 作者 Chollet 在 TPU 上拿 KerasHub+JAX 玩 LLM 后训练，说可能开源，想跟进后训练工具链的可以蹲一下。

#KerasHub #JAX #TPU #微调

官方账号François Chollet@fchollet原文

10月5日

论文 Agent 与开发者19:06

Double-Stitch：免模拟方法学习 Wasserstein 空间拉格朗日力学

Basis Research 提出的 Double-Stitch 免去训练时反复跑数值求解器，比 WLM 快 4-14 倍，还覆盖梯度流搞不定的周期性动力学，JAX 代码已开源。

#Double-Stitch #Wasserstein梯度流 #JAX #单细胞动力学

arXiv cs.LG@Fedor Sergeev 等 7 人原文

9月30日

论文 Agent 与开发者10:33

Jaxolotl：LTL多任务RL统一基准套件

Jaxolotl基准套件让多任务RL研究有了统一标准，训练速度提升220倍，还能对比不同算法的优缺点。

#Jaxolotl #LTL #多任务RL #强化学习

arXiv cs.AI@Mathias Jackermeier 等 3 人原文

9月4日

论文73°11:14

可微分混合建模优化化学传输过程

这篇论文提出了一个结合物理模型和神经网络的混合框架，能从实验数据自动发现本构定律，比传统黑盒模型更具物理一致性。

#JAX #种群平衡方程 #可微分建模 #化学工程

arXiv cs.LG@Arthur Jessop 等 4 人原文

9月3日

论文多源确认精选73°12:09

GRADSOLVE：GPU上ODE集合的快速精确梯度计算

GRADSOLVE解决了ODE集合在GPU上求解与反向微分速度不匹配的问题，大幅提升计算效率。

#GRADSOLVE #JAX #ODE #GPU

事件专题

arXiv cs.LG@Alessio Spurio Mancini8 个信源在谈原文

7月26日

模型官方一手03:08

Open Dreamer 发布：基于 JAX/Flax 的 Dreamer 4 世界模型开源复现，含完整训练配方

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

#Open Dreamer #Dreamer 4 #JAX #Flax

官方一手marktechpost@Asif Razzaq原文

7月2日

产品多源确认15:25

使用Fable 5为RL生物添加毛发和神经网络语音

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

#Fable 5 #Three.js #JAX #MuJoCo

事件专题

AI Will@FinanceYF511 个信源在谈原文

模型多源确认精选15:20

用Fable 5为RL生物添加毛发和神经网络激活值生成的声音

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

#Fable 5 #RL #JAX #MuJoCo

事件专题

AI Will@FinanceYF511 个信源在谈原文

6月19日

模型09:55

CRAX: 基于JAX的快速安全强化学习基准测试，提速约100倍

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

#CRAX #MuJoCo #JAX #安全RL

arXiv cs.LG@Tristan Tomilin 等 4 人原文

5月13日

产品官方一手00:33

MaxText新增单主机TPU的SFT与RL功能

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

#微调/RL #TPU #JAX #MaxText

官方一手Google Developers Blog原文