traveler.core.typing

Functions

get_hinted_shape(annotation)

Extract named dimensions and minimum rank from a jaxtyping annotation.

Classes

Categorical()
CategoricalMeta
CategoricalType()
TypeMaker(shape[, array_type])	A class to create a type for JAX arrays with a specific shape and dtype.
TypeMakerV([array_type])	A class to create a type for JAX arrays with a specific shape and dtype.