Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
Repository Index
76,855 visible repositories in the index
1–4 of 4 · Most stars
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
- differential-equations
- jax
- deep-learning
- neural-differential-equations
Algebraic Multigrid Solvers in Python
- amg
- solver
- python
- multigrid
An automatic code generator for nonlinear model predictive control (NMPC) and the continuation/GMRES method (C/GMRES) based numerical solvers for NMPC
- model-predictive-control
- nonlinear-model-predictive-control
- mpc
- nmpc