4.1.1.4 pde.backends.jax.typing module

Provides support for mypy type checking of the module.

JaxOperatorImplType
JaxDataSetter
JaxGhostCellSetter
JaxVirtualPointEvaluator
JaxInnerStepperType	General backend-level stepping-function type working with jax arrays.

class JaxDataSetter(*args, **kwargs)

Bases: Protocol

class JaxGhostCellSetter(*args, **kwargs)

Bases: Protocol

class JaxInnerStepperType(*args, **kwargs)

Bases: Protocol

General backend-level stepping-function type working with jax arrays.

type JaxOperatorImplType = _OperatorImplStreamBareType[Array]

class JaxVirtualPointEvaluator(*args, **kwargs)

Bases: Protocol