traveler.core#
Core data structures, typing, validation, hashing, and execution utilities.
Submodules#
Array reshaping, lookup, indexing, and pytree concatenation utilities. |
|
JAX-compatible categorical arrays and their low-level annotation syntax. |
|
Utilities for mapping child identifiers to positions in parent identifiers. |
|
Exception and warning types raised by Traveler. |
|
Stable hashing and reproducible JAX pseudorandom-key utilities. |
|
Decorator to create lazy attributes on AttrPyTree instances. |
|
Chunked, multithreaded execution helpers for JAX model operations. |
|
Mutable parameter mappings with safe copy and replacement operations. |
|
Factories for shaped arrays and named categorical-family annotations. |