For some reason, if we replace
init_smc = {"coefficient0_log__": np.full((num_draws, num_chains), 10),
"coefficient1_log__": np.full((num_draws, num_chains), 10)}
with randomized initial values
init_smc = {"coefficient0_log__": np.random.normal(size=(num_draws, num_chains)),
"coefficient1_log__": np.random.normal(size=(num_draws, num_chains))}
the warning is suppressed, even though the shape of the numpy arrays inside init_smc remains the same.