traveler.core#

Core data structures, typing, validation, hashing, and execution utilities.

Submodules#

array_tools

Array reshaping, lookup, indexing, and pytree concatenation utilities.

categorical

JAX-compatible categorical arrays and their low-level annotation syntax.

digitize

Utilities for mapping child identifiers to positions in parent identifiers.

errors

Exception and warning types raised by Traveler.

hashing

Stable hashing and reproducible JAX pseudorandom-key utilities.

lazy

Decorator to create lazy attributes on AttrPyTree instances.

multithread

Chunked, multithreaded execution helpers for JAX model operations.

parameters

Mutable parameter mappings with safe copy and replacement operations.

typing

Factories for shaped arrays and named categorical-family annotations.