diffrax: 一個基於 JAX 的可自動微分與 GPU 加速的數值微分方程求解器函式庫

What it solves

Diffrax 提供了在 JAX 生態系統中數值求解微分方程的方法——特別是常微分方程 (ODE)、隨機微分方程 (SDE) 與控制微分方程 (CDE)——。它使這些求解器能夠進行自動微分，並具備在 GPU 上運行的能力，這對於將物理模型與機器學習整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合整合- 在 JAX 生態系統中提供 ODE, SDE, 與 CDE 求解器。

• Fully 적으로- 根據 J-
• Vmappable operations across the library.
• Support for neural differential equations.
• Includes a diverse range of solvers, including symplectic and implicit options.
• Fully autodifferentiable and GPU-compatible. matrix- 在 JAX 生態系統-
• Fully 적으로- 根據 J-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-
• Vn-