traveler.core.typing.TypeMaker#

class traveler.core.typing.TypeMaker(shape, array_type=jt.Array)[source]#

Bases: object

Create JAX array annotations with a common symbolic shape.

Parameters:

shape (str) – Jaxtyping-compatible symbolic shape shared by generated annotations.
array_type (type, optional) – Array implementation used in generated annotations.

Attributes#

`Categorical`	Bind `TypeMaker.Categorical` to the maker's configured shape.
`Int8`
`Int16`
`Int32`
`Int64`
`Float32`
`Float64`
`Bool`