Skip to content

Commit 1aac2e4

Browse files
Thibault Monselthibmonsel
Thibault Monsel
authored andcommitted
Added first DDE version
1 parent 23d41aa commit 1aac2e4

23 files changed

+12251
-49
lines changed

diffrax/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
)
88
from .autocitation import citation, citation_rules
99
from .brownian import AbstractBrownianPath, UnsafeBrownianPath, VirtualBrownianTree
10+
from .delays import Delays
1011
from .event import (
1112
AbstractDiscreteTerminatingEvent,
1213
DiscreteTerminatingEvent,

diffrax/adjoint.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,8 @@
1313

1414
from .ad import implicit_jvp
1515
from .heuristics import is_sde, is_unsafe_sde
16-
from .saveat import SaveAt, SubSaveAt, save_y
17-
from .solver import (AbstractItoSolver, AbstractRungeKutta,
18-
AbstractStratonovichSolver)
16+
from .saveat import save_y, SaveAt, SubSaveAt
17+
from .solver import AbstractItoSolver, AbstractRungeKutta, AbstractStratonovichSolver
1918
from .term import AbstractTerm, AdjointTerm
2019

2120

@@ -532,6 +531,7 @@ def _loop_backsolve_bwd(
532531
max_steps,
533532
throw,
534533
init_state,
534+
y0_history,
535535
):
536536

537537
#

0 commit comments

Comments
 (0)