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

Implement the IAS15 solver Open

feature

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

patrick-kidger/diffrax#437 · 3 comments · Opened Jun 9, 2024
How to evaluate derivatives of diffrax-solved, equilibrated functions? Open

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

patrick-kidger/diffrax#432 · 4 comments · Opened Jun 3, 2024
Difficulty installing diffrax on apple silicon Open

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

patrick-kidger/diffrax#429 · 8 comments · Opened May 30, 2024
Possible performance issue with implicit solvers Open

question

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

patrick-kidger/diffrax#428 · 2 comments · Opened May 30, 2024
Support for hybrid (mixed discrete-continuous) systems Open

question

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

patrick-kidger/diffrax#423 · 5 comments · Opened May 20, 2024
SDE tests failing with complex inputs Open

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

patrick-kidger/diffrax#422 · 5 comments · Opened May 20, 2024
Problem for adding gradient penalty into sde-gan Open

question

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

patrick-kidger/diffrax#419 · 2 comments · 1 reaction · Opened May 18, 2024
Possible issue with ReversibleHeun solver instability Open

question

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

patrick-kidger/diffrax#417 · 1 comment · Opened May 16, 2024
Inconsistency between constant and adaptive step size solvers with Discrete terminating events Open

question

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

patrick-kidger/diffrax#416 · 1 comment · Opened May 13, 2024
Work-around for explicit DAE system? Open

question

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

patrick-kidger/diffrax#413 · 3 comments · Opened May 8, 2024
Sharding integration is much slower than pmap Open

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

patrick-kidger/diffrax#407 · 22 comments · Opened Apr 21, 2024
[Feature request] A delay differential equations solver Open

feature

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

patrick-kidger/diffrax#406 · 10 comments · Opened Apr 20, 2024
Using diffrax to integrate system dynamics for Model Predictive Control Open

question

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

patrick-kidger/diffrax#393 · 4 comments · Opened Mar 24, 2024
Complex input in diffeqsolve with PIDController Open

bug question

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

patrick-kidger/diffrax#389 · 3 comments · Opened Mar 15, 2024
Complex support for implicit solver Open

question

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

patrick-kidger/diffrax#388 · 3 comments · Opened Mar 15, 2024
Explosion of steps for specific parameter values Open

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

patrick-kidger/diffrax#386 · 12 comments · Opened Mar 13, 2024
Jacobian matrix for ODE solvers Open

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

patrick-kidger/diffrax#375 · 2 comments · Opened Feb 16, 2024
Save num_steps At ts Open

question

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

patrick-kidger/diffrax#374 · 5 comments · Opened Feb 16, 2024
Support Lineax operators for the vector field of ControlTerm Open

feature

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

patrick-kidger/diffrax#370 · 4 comments · Opened Feb 10, 2024
Batch training of NODE with varying external input (forcing) per batch element Open

question

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

patrick-kidger/diffrax#365 · 4 comments · Opened Jan 31, 2024
Zero gradient when using jnp.piecewise inside an ODE Open

question

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

patrick-kidger/diffrax#363 · 2 comments · Opened Jan 29, 2024
What is the best way to integrate a function? Open

question

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

patrick-kidger/diffrax#361 · 5 comments · Opened Jan 26, 2024
Parametric type hinting for control paths Open

refactor

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

patrick-kidger/diffrax#359 · 1 comment · Opened Jan 24, 2024
Solving simple dynamics: ControlTerm piecewise product Open

question

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

patrick-kidger/diffrax#358 · 3 comments · Opened Jan 22, 2024
Vector of Wiener processes Open

question

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

patrick-kidger/diffrax#355 · 2 comments · Opened Jan 12, 2024
Passing additional state between subsequent RHS evaluations? Open

feature next

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

patrick-kidger/diffrax#343 · 2 comments · Opened Dec 21, 2023
Fitting ODE model with `diffeqsolve` is extremely slow using NUTS on GPU Open

question

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

patrick-kidger/diffrax#338 · 3 comments · 2 reactions · Opened Dec 8, 2023
inf values after triggering event function. Open

question

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

patrick-kidger/diffrax#335 · 4 comments · Opened Nov 29, 2023
Simulate SDE with white noise? Open

question

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

patrick-kidger/diffrax#334 · 3 comments · Opened Nov 28, 2023
Frequent JIT-recompile of discrete_terminating_event Open

question

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

patrick-kidger/diffrax#333 · 2 comments · Opened Nov 18, 2023
Taking more than one gradient fails with default RecursiveCheckpointAdjoint Open

refactor

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

patrick-kidger/diffrax#332 · 2 comments · Opened Nov 9, 2023
replace_nans_at_start overwrite existing values Open

bug

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

patrick-kidger/diffrax#324 · 1 comment · Opened Oct 16, 2023
Treating Channels that only have missing values in interpolations Open

bug

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

patrick-kidger/diffrax#323 · 1 comment · Opened Oct 11, 2023
Solver for very stiff Neural Ordinary Differential Equations Open

question

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

patrick-kidger/diffrax#319 · 7 comments · 2 reactions · Opened Oct 5, 2023
diffeqsolve time parameters Open

question

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

patrick-kidger/diffrax#316 · 1 comment · Opened Oct 2, 2023
Step-by-Step Interface is very Inefficient vs. Base diffeqsolve Open

question

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

patrick-kidger/diffrax#313 · 2 comments · Opened Sep 17, 2023
`jax.clear_caches` after backpropagating through `diffeqsolve` causes segmentation fault Open

question

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

patrick-kidger/diffrax#306 · 1 comment · Opened Sep 9, 2023
Multiple shooting Open

feature next

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

patrick-kidger/diffrax#302 · 10 comments · Opened Aug 29, 2023
SaveAt dense and fn Open

refactor

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

patrick-kidger/diffrax#301 · 5 comments · Opened Aug 29, 2023
Making sure that the backward pass is using Algorithm-2 from the paper 'Efficient and Accurate Gradients for Neural SDEs' Open

feature

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

patrick-kidger/diffrax#300 · 3 comments · Opened Aug 29, 2023
NewtonNonlinearSolver returns nan when initialized with root value Open

bug

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

patrick-kidger/diffrax#296 · 1 comment · Opened Aug 24, 2023
How to use DiscreteTerminatingEvent to terminate integration as soon as any ODEterm value becomes NaN? Open

question

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

patrick-kidger/diffrax#290 · 6 comments · 1 reaction · Opened Aug 10, 2023
Custom VJP error with Custom PyTrees and RecursiveCheckpoint Open

question

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

patrick-kidger/diffrax#287 · 3 comments · Opened Aug 4, 2023
Ensemble simulations of small ODEs on GPUs Open

question

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

patrick-kidger/diffrax#281 · 5 comments · Opened Jul 23, 2023
Inspection of gradient calculation Open

question

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

patrick-kidger/diffrax#280 · 3 comments · Opened Jul 21, 2023
METAL Backend Support Open

question

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

patrick-kidger/diffrax#279 · 5 comments · Opened Jul 15, 2023
Allowing for state change after each step Open

feature

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

patrick-kidger/diffrax#277 · 1 comment · Opened Jul 12, 2023
safe_zip() error when passing arrays into args Open

question

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

patrick-kidger/diffrax#271 · 5 comments · Opened Jun 24, 2023
Making diffrax Haiku compliant Open

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

patrick-kidger/diffrax#270 · 3 comments · Opened Jun 22, 2023
Support for modified Semi-implicit euler solver Open

feature

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

patrick-kidger/diffrax#269 · 2 comments · Opened Jun 22, 2023

patrick-kidger/diffrax

Get new issues in your inbox