state

JAX visitors for state/time expressions.

Visitors: State, Time, Tau

Lowers a State leaf to a JAX function that slices the corresponding entries out of the unified state vector x, using the slice assigned during unification. Time is a State subclass and shares the same visitor; lowering a state whose slice is still unset is a usage error and raises. Tau — the normalized node coordinate used by initial-guess expressions — reads its value off the node grid carried in params.