fdtdx.typing

fdtdx.typing.RealShape3D module-attribute

RealShape3D = tuple[float, float, float]

3D shape with real-valued (physical) dimensions in meters.

fdtdx.typing.PartialRealShape3D module-attribute

PartialRealShape3D = tuple[
    OptionalAxisSize, OptionalAxisSize, OptionalAxisSize
]

Partial 3D shape where some physical dimensions may be undefined (None).

fdtdx.typing.GridShape3D module-attribute

GridShape3D = tuple[int, int, int]

3D shape with integer dimensions in grid points.

fdtdx.typing.PartialGridShape3D module-attribute

PartialGridShape3D = tuple[
    OptionalGridAxisSize,
    OptionalGridAxisSize,
    OptionalGridAxisSize,
]

Partial 3D grid shape where some dimensions may be undefined (None).

fdtdx.typing.BackendOption module-attribute

BackendOption = Literal['gpu', 'tpu', 'cpu', 'METAL']

Backend options for JAX. Can be either gpu, tpu, cpu or METAL