diffrax/diffrax/__init__.py at 49acb23d8d1e8a43a0bea3254a33535db65d018d · patrick-kidger/diffrax · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
from .adjoint import (
    AbstractAdjoint,
    BacksolveAdjoint,
    DirectAdjoint,
    ImplicitAdjoint,
    RecursiveCheckpointAdjoint,
)
from .autocitation import citation, citation_rules
from .brownian import AbstractBrownianPath, UnsafeBrownianPath, VirtualBrownianTree
from .event import (
    AbstractDiscreteTerminatingEvent,
    DiscreteTerminatingEvent,
    SteadyStateEvent,
)
from .global_interpolation import (
    AbstractGlobalInterpolation,
    backward_hermite_coefficients,
    CubicInterpolation,
    DenseInterpolation,
    linear_interpolation,
    LinearInterpolation,
    rectilinear_interpolation,
)
from .integrate import diffeqsolve
from .local_interpolation import (
    AbstractLocalInterpolation,
    FourthOrderPolynomialInterpolation,
    LocalLinearInterpolation,
    ThirdOrderHermitePolynomialInterpolation,
)
from .misc import adjoint_rms_seminorm
from .nonlinear_solver import (
    AbstractNonlinearSolver,
    AffineNonlinearSolver,
    NewtonNonlinearSolver,
    NonlinearSolution,
)
from .path import AbstractPath
from .saveat import SaveAt, SubSaveAt
from .solution import is_event, is_okay, is_successful, RESULTS, Solution
from .solver import (
    AbstractAdaptiveSolver,
    AbstractDIRK,
    AbstractERK,
    AbstractESDIRK,
    AbstractImplicitSolver,
    AbstractItoSolver,
    AbstractRungeKutta,
    AbstractSDIRK,
    AbstractSolver,
    AbstractStratonovichSolver,
    AbstractWrappedSolver,
    Bosh3,
    ButcherTableau,
    CalculateJacobian,
    Dopri5,
    Dopri8,
    Euler,
    EulerHeun,
    HalfSolver,
    Heun,
    ImplicitEuler,
    ItoMilstein,
    KenCarp3,
    KenCarp4,
    KenCarp5,
    Kvaerno3,
    Kvaerno4,
    Kvaerno5,
    LeapfrogMidpoint,
    Midpoint,
    MultiButcherTableau,
    Ralston,
    ReversibleHeun,
    SemiImplicitEuler,
    StormerVerlet,
    Sil3,
    StratonovichMilstein,
    Tsit5,
)
from .step_size_controller import (
    AbstractAdaptiveStepSizeController,
    AbstractStepSizeController,
    ConstantStepSize,
    PIDController,
    StepTo,
)
from .term import (
    AbstractTerm,
    ControlTerm,
    MultiTerm,
    ODETerm,
    WeaklyDiagonalControlTerm,
)


__version__ = "0.4.1"