diffrax and exponax
Diffrax provides general-purpose ODE/SDE solvers that form the computational foundation upon which specialized PDE solvers like Exponax build their domain-specific discretization schemes.
About diffrax
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
This is a tool for scientists and engineers who need to simulate complex systems over time using differential equations. It takes your system's equations and initial conditions, then precisely calculates how the system evolves. Researchers in fields like physics, biology, and finance can use this to model dynamic processes.
About exponax
Ceyron/exponax
Efficient Differentiable n-d PDE Solvers in JAX.
This project helps researchers and engineers quickly simulate and analyze complex physical systems described by partial differential equations (PDEs) in one, two, or three dimensions. You provide the equation's parameters and initial conditions, and it calculates how the system evolves over time. It's ideal for scientists, physicists, and engineers working on fluid dynamics, material science, or reaction-diffusion processes.
Related comparisons
Scores updated daily from GitHub, PyPI, and npm data. How scores work