Diffrax 是基于 JAX 的数值微分方程解析库，适用于常微分方程、随机微分方程和受控微分方程的求解。其特点包括多种解析器选择（如 Tsit5、Dopri8、辛解析器、隐式解析器）、使用 PyTree 作为状态存储、支持稠密解和多种反向传播方法，并支持神经微分方程。兼容 Python 3.9+、JAX 0.4.13+ 和 Equinox 0.10.11+。