patrick-kidger/diffrax

View on GitHub

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

Stars: 2.1k
Forks: 189
Open beginner issues: 0
Indexed issues: 215
Avg merge: 3d 18h
Merged PRs (30d): 1

Dominant language: Python
License: Apache-2.0
Last GitHub push: Sep 6, 2026
Latest indexed: Sep 20, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Beginner friendly Everything

Open All

215 open issues indexed

Question: Solving an SDE by applying an ODE Runge-Kutta method to the Stratonovitch form in Diffrax Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/diffrax#769 · 1 comment · Opened Sep 14, 2026
Zero-order-hold inputs with fixed-step solvers in a single diffeqsolve Open

Difficulty 4/5 3-5 days Newbie friendliness 52/100

patrick-kidger/diffrax#766 · 4 comments · Opened Aug 13, 2026
problems when calling diffeqsolve inside a blackjax sampler with *shard_map*. Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/diffrax#765 · 6 comments · Opened Aug 10, 2026
How to sample fresh noise at each step for a non-linear noise-perturbed ODE ``dX = f(X, z)dt``? Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/diffrax#759 · 8 comments · Opened Jun 10, 2026
Documentation and implementation of ConstantStepSize do not quite align Open

Difficulty 3/5 1-2 days Newbie friendliness 52/100

patrick-kidger/diffrax#757 · Opened May 27, 2026
Enforcing Absolute Unitarity in Differentiable Physics: The Float-to-Integer Ledger Approach Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

patrick-kidger/diffrax#750 · Opened Apr 30, 2026
Leave some elements of `jax` array unchanged during `diffeqsolve` without computation Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#747 · 1 comment · Opened Apr 16, 2026
Low Storage Methods Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#743 · 3 comments · Opened Mar 17, 2026
Return value of the vector field at t0 without recomputation Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#741 · 2 comments · Opened Mar 13, 2026
Running diffeqsolve for a fixed number of *accepted* steps Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#740 · 1 comment · Opened Mar 4, 2026
Support for Sharding 2.0 2.0 Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/diffrax#738 · 2 comments · Opened Feb 26, 2026
Gradients of state variables in post-event simulations Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#729 · 15 comments · Opened Feb 11, 2026
Creating Support for Delay Differential Equations Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/diffrax#721 · 3 comments · Opened Jan 20, 2026
auto-parallelization alongside `vmap` is very slow Open

question

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/diffrax#718 · 4 comments · Opened Jan 12, 2026
Kvaerno, KenCarp solvers fail jax tracing for a multicomponent state and jax 0.8.2 Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/diffrax#717 · 1 comment · Opened Jan 12, 2026
Old versions much faster? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#711 · 4 comments · Opened Dec 2, 2025
(Non-constant) mass matrix example Open

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/diffrax#710 · 1 comment · Opened Dec 2, 2025
JIT compiling VirtualBrownianTree.evaluate() hangs indefinitely when using jnp.int32 to pass shape to ShapeDtypeStruct Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#707 · 5 comments · Opened Nov 13, 2025
Obscure error for passing EulerHeun instead of EulerHeun() Open

Difficulty 3/5 1-2 days Newbie friendliness 42/100

patrick-kidger/diffrax#705 · 3 comments · Opened Nov 7, 2025
Infinite loop in PID stepsize controller due to float precision error Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/diffrax#703 · 3 comments · Opened Nov 5, 2025
Code breaks on an oscillator-like problem Open

Difficulty 4/5 3-5 days Newbie friendliness 28/100

patrick-kidger/diffrax#702 · 2 comments · Opened Nov 1, 2025
Cannot reproduce example from stiff ode Open

question

Difficulty 3/5 1-2 days Newbie friendliness 25/100

patrick-kidger/diffrax#700 · 1 comment · Opened Oct 29, 2025
Support Events for BacksolveAdjoint Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#698 · 2 comments · Opened Oct 24, 2025
How to enforce non-negativity constraints? Open

question

Difficulty 5/5 Over a week Newbie friendliness 30/100

patrick-kidger/diffrax#697 · 2 comments · Opened Oct 5, 2025
Out of Memory: How to reduce the memory cost for calculating the derivatives wrt the input Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#691 · 2 comments · Opened Sep 16, 2025
Our of memory : How to optimize the memory use in Diffrax framework? Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/diffrax#685 · 4 comments · Opened Sep 8, 2025
FYI: JAX 0.7.0 and 0.7.1 are known to be buggy with Diffrax. Open

Difficulty 1/5 Under an hour Newbie friendliness 25/100

patrick-kidger/diffrax#680 · 2 comments · Opened Aug 21, 2025
Implicit SDE Solvers? Open

feature

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/diffrax#676 · 1 comment · Opened Aug 8, 2025
ODE solver: some time period has algebraic solution Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#673 · 4 comments · Opened Aug 3, 2025
Question about an “AllSaveAdjoint” (store-all-steps, no recomputation) variant Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#668 · 5 comments · Opened Jul 20, 2025
sol.ts contains wrong values in some cases Open

Difficulty 4/5 3-5 days Newbie friendliness 42/100

patrick-kidger/diffrax#663 · 1 comment · Opened Jul 7, 2025
Implementation of multirate methods in diffrax Open

feature

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#661 · 2 comments · Opened Jul 2, 2025
How to pass kwargs to `root_find` Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/diffrax#658 · 6 comments · Opened Jun 21, 2025
Questions on NeuralCDEs Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

patrick-kidger/diffrax#653 · Opened Jun 13, 2025
Recommendation for real-time publishing of vector field Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#651 · 2 comments · Opened Jun 3, 2025
Recommendation for saving auxilary data Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#650 · 2 comments · Opened Jun 3, 2025
Specification of sign change direction Open

Difficulty 4/5 3-5 days Newbie friendliness 32/100

patrick-kidger/diffrax#640 · 3 comments · Opened May 23, 2025
Handling of multiple Events triggered during one integration step Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/diffrax#639 · 7 comments · Opened May 23, 2025
Endless loop after jax.jacrev(ode solver)(args) Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#638 · 2 comments · Opened May 22, 2025
How to properly scale with multiple event conditions? Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/diffrax#637 · 7 comments · Opened May 22, 2025
Steady state solver termination Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#635 · 8 comments · Opened May 21, 2025
Could you give an example of interactively step through a implicit solver? Open

feature question

Difficulty 3/5 1-2 days Newbie friendliness 45/100

patrick-kidger/diffrax#633 · 6 comments · Opened May 21, 2025
Vmapped jnp.interp throws error when debugging nan's with disabled jit Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#630 · 7 comments · Opened May 8, 2025
Adjoint hessian method Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/diffrax#628 · 7 comments · Opened May 5, 2025
os.environ['XLA_FLAGS'] = '--xla_cpu_use_thunk_runtime=false' no longer working with Jax 0.6.0 Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/diffrax#627 · 3 comments · Opened May 2, 2025
Type Error in Neural ODE Example Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/diffrax#622 · 11 comments · Opened Apr 26, 2025
Add support for Jax.experimental.jet. Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#616 · 2 comments · 1 reaction · Opened Apr 7, 2025
Different results for dopri5 depending on loop vs solve Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/diffrax#613 · 4 comments · Opened Mar 31, 2025
Question about different adjoint methods Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/diffrax#611 · 3 comments · Opened Mar 28, 2025
Solver selection -- multistep/Adams methods Open

feature

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/diffrax#609 · 4 comments · Opened Mar 23, 2025

patrick-kidger/diffrax

Get new issues in your inbox