patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Pythonmachine-learningdeep-learningneural-networksdynamical-systemsdifferential-equationsequinoxjaxneural-differential-equations
This is stars and forks stats for /patrick-kidger/diffrax repository. As of 19 Apr, 2024 this repository has 1017 stars and 89 forks.

Diffrax Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. Diffrax is a JAX-based library providing numerical differential equation solvers. Features include: ODE/SDE/CDE (ordinary/stochastic/controlled) solvers; lots of different solvers (including Tsit5, Dopri8, symplectic solvers, implicit solvers); vmappable everything (including the region of integration); using a PyTree as the state; dense solutions; multiple adjoint methods for backpropagation; support for...
Read on GithubGithub Stats Page
repotechsstarsweeklyforksweekly
matrix-org/synapsePythonRustHTML11.4k02.1k0
apache/airflowPythonTypeScriptShell32k+9413k+30
PyCQA/isortPythonOther6k05770
jordansissel/fpmRubyShellHTML10.9k01.1k0
dair-ai/ML-YouTube-Courses11.4k+341.4k+3
AdaCore/ada_language_serverAdaTypeScriptPython2010560
ablab/quastAMPLC++Perl333+1720
antlr/grammars-v4ANTLRJavaPython9.3k+203.6k0
lnbook/lnbookAsciiDocShellHTML2.7k06480
bitcoinbook/bitcoinbookAsciiDocPythonHTML21.8k+155.8k-3