kd.typing.Memo

class kauldron.typing.Memo(

single: dict[str, int],

variadic: dict[str, tuple[int, ...]],

)

Bases: object

Jaxtyping information about the shapes in the current scope.

single: dict[str, int]

variadic: dict[str, tuple[int, ...]]

classmethod from_current_context()

Create a Memo from the current typechecking context.