PyRank

Neural Differential Equations Python Packages

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

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

1.2M 2K 231
patrick-kidger
diffrax

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

264K 2K 194
patrick-kidger
torchcde

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

63K 494 50
GalacticDynamics
diffraxtra

Extras for Diffrax: OOP and vectorization

12K 6 3
scDiffEq
scdiffeq

Learn single-cell dynamics using neural differential equations

1K 72 13
samholt
torchlaplace

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

344 86 12
RunnersNum40
lerax

Reinforcement Learning in Jax

176 6 0