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

Model is slower using diffrax, and even more when calling a function inside the vector_field Open

question

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

patrick-kidger/diffrax#605 · 6 comments · Opened Mar 18, 2025
Speeding Up Evaluation of Padded Time Series Using Events? Open

question

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

patrick-kidger/diffrax#602 · 5 comments · Opened Mar 13, 2025
BacksolveAdjoint() fails to compute gradients Open

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

patrick-kidger/diffrax#596 · 4 comments · Opened Mar 3, 2025
Passing a state between vector_field calls Open

question

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

patrick-kidger/diffrax#595 · 10 comments · Opened Feb 25, 2025
3D heat diffusion Open

question

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

patrick-kidger/diffrax#594 · 3 comments · Opened Feb 24, 2025
Significant performance difference: diffeqsolve vs. lax.scan - Expected Behavior? Open

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

patrick-kidger/diffrax#592 · 4 comments · Opened Feb 17, 2025
Safety of taking a delta t = 0 step? Open

question

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

patrick-kidger/diffrax#591 · 2 comments · Opened Feb 15, 2025
Clarification of error message Open

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

patrick-kidger/diffrax#590 · 1 comment · Opened Feb 13, 2025
Typeguard old version pin Open

question

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

patrick-kidger/diffrax#589 · 1 comment · Opened Feb 13, 2025
Interest in folding in diffraxtra Open

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

patrick-kidger/diffrax#585 · 1 comment · Opened Feb 4, 2025
[Proposal]: Add more Ito SDE Solvers Open

feature

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

patrick-kidger/diffrax#576 · 2 comments · Opened Jan 28, 2025
[Question] odd behaviour when solving SDE for differential initial conditions Open

question

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

patrick-kidger/diffrax#560 · 2 comments · Opened Jan 5, 2025
Adjoints question Open

question

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

patrick-kidger/diffrax#558 · 8 comments · Opened Jan 4, 2025
Question: DirectAdjoint is faster than RecursiveCheckpointAdjoint? Open

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

patrick-kidger/diffrax#549 · 20 comments · Opened Jan 1, 2025
sparse jacobian solve Open

question

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

patrick-kidger/diffrax#545 · 4 comments · Opened Dec 17, 2024
jax.experimental.enable_x64 and jit Open

question

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

patrick-kidger/diffrax#523 · 1 comment · Opened Nov 12, 2024
[Question] Neural CDE for regression Open

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

patrick-kidger/diffrax#519 · 3 comments · Opened Nov 4, 2024
[Regression] Slower integration of differential equations since jaxlib > 0.4.32.dev20240807 Open

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

patrick-kidger/diffrax#518 · 4 comments · Opened Oct 25, 2024
Performance issue with SDE solver Open

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

patrick-kidger/diffrax#517 · 27 comments · Opened Oct 21, 2024
JaxStackTraceBeforeTransformation error with parametrized ODE Open

question

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

patrick-kidger/diffrax#513 · 4 comments · Opened Oct 11, 2024
`EQX_ON_ERROR=breakpoint` and `diffeqsolve` causes tracer leakage ? Open

question

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

patrick-kidger/diffrax#512 · 2 comments · Opened Oct 10, 2024
Calculating First-order Sensitivity using diffrax Open

question

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

patrick-kidger/diffrax#511 · 1 comment · Opened Oct 8, 2024
Event and PIDController: event doesn't always occure Open

question

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

patrick-kidger/diffrax#507 · 3 comments · Opened Oct 4, 2024
[time-scale separation] Adaptive step size for different variables? Open

question

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

patrick-kidger/diffrax#505 · 1 comment · Opened Sep 30, 2024
Advice about time-based 'events' Open

question

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

patrick-kidger/diffrax#504 · 2 comments · Opened Sep 28, 2024
Incorrect gradient in toy adaptive ODE Open

documentation

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

patrick-kidger/diffrax#499 · 4 comments · Opened Aug 24, 2024
Restart after event Open

question

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

patrick-kidger/diffrax#492 · 2 comments · 1 reaction · Opened Aug 17, 2024
Why can't UBP be backproped? Open

question

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

patrick-kidger/diffrax#490 · 7 comments · Opened Aug 16, 2024
VBT vs brownian path slowdown Open

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

patrick-kidger/diffrax#489 · 7 comments · Opened Aug 16, 2024
Intermediate saved values are sometimes `inf` Open

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

patrick-kidger/diffrax#488 · 3 comments · 1 reaction · Opened Aug 16, 2024
Why are step_ts and jump_ts treated differently here? Open

question

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

patrick-kidger/diffrax#483 · 10 comments · Opened Aug 10, 2024
Feature Request: Complex-Valued Integration With ZVODE - CVODE in Jax (autodiff) Open

question

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

patrick-kidger/diffrax#477 · 5 comments · 1 reaction · Opened Aug 6, 2024
Use values computed in ODETerms inside of SaveAt Open

question

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

patrick-kidger/diffrax#476 · 5 comments · Opened Aug 5, 2024
Additive SDE throws error with SRK style solvers Open

question

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

patrick-kidger/diffrax#474 · 8 comments · Opened Aug 1, 2024
Making subsaveat consider previous saves and not just the current one Open

question

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

patrick-kidger/diffrax#472 · 4 comments · Opened Jul 31, 2024
Adaptive SDE Docs Open

question

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

patrick-kidger/diffrax#471 · 1 comment · Opened Jul 31, 2024
Understand diffrax.diffeqsolve Open

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

patrick-kidger/diffrax#467 · 1 comment · Opened Jul 24, 2024
Accelerate ODE solver [What did I miss?] Open

question

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

patrick-kidger/diffrax#466 · 3 comments · Opened Jul 23, 2024
RecursiveCheckpointAdjoint not working for two-level minimisation Open

question

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

patrick-kidger/diffrax#465 · 2 comments · Opened Jul 20, 2024
Diffrax activating the event on a rejected step Open

bug

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

patrick-kidger/diffrax#464 · 11 comments · Opened Jul 18, 2024
statefully evolving an auxiliary variable Open

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

patrick-kidger/diffrax#462 · 20 comments · 1 reaction · Opened Jul 12, 2024
Coupled SDE System Implementation Open

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

patrick-kidger/diffrax#461 · 7 comments · Opened Jul 10, 2024
Demos not working Open

question

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

patrick-kidger/diffrax#459 · 2 comments · Opened Jul 7, 2024
DAE status Open

feature

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

patrick-kidger/diffrax#457 · 1 comment · Opened Jul 4, 2024
SDE - share computation common to drift and diffusion terms Open

question

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

patrick-kidger/diffrax#455 · 5 comments · Opened Jul 2, 2024
Steady-state event occurs immediately if state with large value constant Open

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

patrick-kidger/diffrax#448 · 7 comments · Opened Jun 26, 2024
Making a simulation simulatneously on 2 different solvers Open

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

patrick-kidger/diffrax#447 · 5 comments · Opened Jun 25, 2024
Can't use Equinox inside `term` Open

question

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

patrick-kidger/diffrax#446 · 4 comments · Opened Jun 24, 2024
Warning for solving example ODE from docu Open

question

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

patrick-kidger/diffrax#445 · 3 comments · 1 reaction · Opened Jun 20, 2024
Support for uncertainty propagation? Open

feature

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

patrick-kidger/diffrax#438 · 2 comments · Opened Jun 9, 2024

patrick-kidger/diffrax

Get new issues in your inbox