4.1.1.4 pde.backends.jax.typing module

Provides support for mypy type checking of the module.

JaxOperatorType	An operator that acts on an array.
JaxDataSetter
JaxGhostCellSetter
JaxVirtualPointEvaluator

class JaxDataSetter(*args, **kwargs)

Bases: Protocol

class JaxGhostCellSetter(*args, **kwargs)

Bases: Protocol

class JaxInnerStepperType(*args, **kwargs)

Bases: Protocol

General stepper type working with jax arrays.

class JaxOperatorType(*args, **kwargs)

Bases: Protocol

An operator that acts on an array.

class JaxVirtualPointEvaluator(*args, **kwargs)

Bases: Protocol