实体
Diffrax
Diffrax
PulseAugur coverage of Diffrax — every cluster mentioning Diffrax across labs, papers, and developer communities, ranked by signal.
总计 · 30天
2
90 天内 2
发布 · 30天
0
90 天内 0
论文 · 30天
2
90 天内 2
层级分布 · 90 天
主题
情绪 · 30 天
1 天有情绪数据
最近 · 第 1/1 页 · 共 2 条
-
新的KPPax工具支持大气化学的可微分机器学习
研究人员开发了KPPax，这是一个新的预处理器，它使得在机器学习框架中使用以Kinetic PreProcessor (KPP)语言编写的大气化学机制成为可能。KPPax生成的代码是可微分的、即时编译的，并使用JAX进行向量化。这使得通过diffrax库的隐式求解器和伴随计算精确的灵敏度，从而促进大气化学中的参数估计和混合物理-机器学习建模。
-
GRADSOLVE库加速GPU上ODE梯度计算
一个名为GRADSOLVE的新型开源JAX库已被开发出来，用于加速NVIDIA GPU上常微分方程（ODE）集成精确梯度的计算。该库解决了现有GPU软件在快速ODE求解和高效梯度计算之间进行权衡的性能瓶颈。GRADSOLVE通过记录求解器步骤然后执行固定步重放来实现更快的梯度计算，与Diffrax等标准方法相比，微分速度显著提高。