Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
2021-07-25
否
2025-07-30T21:09:03Z
该仓库已收录但尚未编辑。项目介绍及使用教程请前往 GitHub 阅读 README
#计算机科学#Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
#计算机科学#Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)