neuralqx.experimental.vqs.mc.mc_state.utils package¶

params_are_complex(params)¶

Return type:: bool

make_grad_qgt_compatible(params, grad)¶: Drop spurious imaginary Monte Carlo noise for real-parameter pytrees.

tree_add(a, b)¶

tree_add_scaled(a, b, scale)¶

tree_scale(a, scale)¶

tree_zeros_like(a)¶

same_treedef(a, b)¶

Return type:: bool

get_stats_mean(stats_obj)¶

Return type:: float