jaxopt.tree\_util.tree\_sum =========================== .. currentmodule:: jaxopt.tree_util .. autofunction:: tree_sum