BayesianSDEsolver
Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.
Install / Use
/learn @ylefay/BayesianSDEsolverREADME
Bayesian SDE solvers
Companion code in JAX to the article preprint: Modelling pathwise uncertainty of Stochastic Differential Equations samplers via Probabilistic Numerics by Yvann Le Fay, Simo Särkkä and Adrien Corenflos.
What is it?
This is a JAX implementation of 1.0 strongly convergent SDE schemes including novel Gaussian-based probabilistic SDE solvers.
Supported features
- Classic SDE schemes: Euler-Maruyama, 1.5 Taylor-Itô
- Exotic Gaussian filtering SDE schemes including 1.0 strongly convergent scheme based on piecewise polynomial approximations of the Brownian motion. Can be used both for pathwise and moment computations.
- Euler ODE scheme.
- Extended Kalman filtering, with lower square root implementation.
Usage
See the scripts and tests folders for examples of usage.
Reproducing the results of the article
Please refer to scripts/README.md for instructions on how to reproduce the results of the article.
License
This project is licensed under the MIT License.
Related Skills
node-connect
349.9kDiagnose OpenClaw node connection and pairing failures for Android, iOS, and macOS companion apps
frontend-design
109.8kCreate distinctive, production-grade frontend interfaces with high design quality. Use this skill when the user asks to build web components, pages, or applications. Generates creative, polished code that avoids generic AI aesthetics.
openai-whisper-api
349.9kTranscribe audio via OpenAI Audio Transcriptions API (Whisper).
qqbot-media
349.9kQQBot 富媒体收发能力。使用 <qqmedia> 标签,系统根据文件扩展名自动识别类型(图片/语音/视频/文件)。
