traveler.core.typing

Factories for shaped arrays and named categorical-family annotations.

Functions

get_hinted_shape(annotation)

Extract named dimensions and minimum rank from a jaxtyping annotation.

Classes

Categorical()	Namespace for annotations tied to named category families.
TypeMaker(shape[, array_type])	Create JAX array annotations with a common symbolic shape.