Advanced#
Most of the following modules are mostly intended for internal use.
typingModulemodel_utilModuletree_utilModuleabs()abs_sqr()abs_sqrt()add()all_close()all_equal()astype()batch_leaves()concatenate()div()extend_structure_from_strpaths()get_shapes()getitem_by_strpath()has_uniform_leaf_shapes()is_shape_leaf()is_single_leaf()make_mutable()mean()mul()neg()pow()pytree_split()random_permute_leaves()safecast()scalar_add()scalar_mul()setitem_by_strpath()shapes_equal()sqrt()strfmt_pytree()sub()uniform_leaf_shapes_equal()
eigen_utilModulegraph_utilModule