JAXopt
Documentation
Basics
Unconstrained optimization
Constrained optimization
Quadratic programming
Non-smooth optimization
Stochastic optimization
Root finding
Fixed point resolution
Nonlinear least squares
Linear system solving
Implicit differentiation
Loss and objective functions
Line search
Perturbed optimization
API
API at a glance
Optimization
Linear system solving
Nonlinear least squares
Root finding
Fixed point resolution
Implicit differentiation
Line search
Perturbed optimizers
Isotonic regression
Tree utilities
jaxopt.tree_util.tree_add
jaxopt.tree_util.tree_sub
jaxopt.tree_util.tree_mul
jaxopt.tree_util.tree_div
jaxopt.tree_util.tree_scalar_mul
jaxopt.tree_util.tree_add_scalar_mul
jaxopt.tree_util.tree_vdot
jaxopt.tree_util.tree_sum
tree_sum()
jaxopt.tree_util.tree_l2_norm
jaxopt.tree_util.tree_zeros_like
Examples
Notebook gallery
Example gallery
About
Authors
Changelog
Source code
Issue tracker
Development
JAXopt
»
API at a glance
»
jaxopt.tree_util.tree_sum
View page source
jaxopt.tree_util.tree_sum
jaxopt.tree_util.
tree_sum
(
tree_x
)
[source]
Compute sum(tree_x).