PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
google-research
torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

2.5M 2K 222
patrick-kidger
diffrax

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

180K 2K 178
patrick-kidger
torchcde

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

67K 482 50
scDiffEq
scdiffeq

Learn single-cell dynamics using neural differential equations

2K 68 13
samholt
torchlaplace

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

2K 82 12
GalacticDynamics
diffraxtra

Extras for Diffrax: OOP and vectorization

1K 6 3
RunnersNum40
lerax

Deep Reinforcement Learning with JAX and Equinox.

361 6 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery