PyRank

Neural Differential Equations Python Packages

Python packages with the GitHub topic neural-differential-equations. Sorted by relevance, with stars and monthly downloads.
google-research
torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

2.2M 2K 223
patrick-kidger
diffrax

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

192K 2K 180
patrick-kidger
torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

59K 482 50
samholt
torchlaplace

Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.

2K 82 12
scDiffEq
scdiffeq

Learn single-cell dynamics using neural differential equations

1K 70 13
GalacticDynamics
diffraxtra

diffrax extras: OOP and vectorization

716 6 3
RunnersNum40
lerax

Reinforcement Learning in Jax

235 6 0