热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#CDE
Diffrax: JAX中强大的微分方程求解器
2 个月前
Diffrax是一个基于JAX的数值微分方程求解库,支持ODE/SDE/CDE等多种微分方程类型,提供多种求解器和自动微分功能,可在GPU上高效运行。
Diffrax
JAX
SDE
ODE
CDE
Github
开源项目
2 个月前
相关项目
diffrax
Diffrax 是基于 JAX 的数值微分方程解析库,适用于常微分方程、随机微分方程和受控微分方程的求解。其特点包括多种解析器选择(如 Tsit5、Dopri8、辛解析器、隐式解析器)、使用 PyTree 作为状态存储、支持稠密解和多种反向传播方法,并支持神经微分方程。兼容 Python 3.9+、JAX 0.4.13+ 和 Equinox 0.10.11+。
查看
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号