EngineBuilder.set_initial_values(model_state, multiple_chains=False)[source]#

Sets the initial model state.

If multiple_chains is true the model_state will be used as is; otherwise model_state will be used as the initial values for each chain. Note that if multiple_chains is true, the first axis of each leaf of model_state refers to the chain.