Index

_

_add_prefix() (in module pybaum.tree_util) _array_element_names() (in module pybaum.registry_entries) _dict() (in module pybaum.registry_entries) _flatten_pandas_dataframe() (in module pybaum.registry_entries) _get_names_pandas_dataframe() (in module pybaum.registry_entries) _index_element_to_string() (in module pybaum.registry_entries) _is_jax_array() (in module pybaum.typecheck) _is_namedtuple() (in module pybaum.typecheck) _jax_array() (in module pybaum.registry_entries) _leaf_names() (in module pybaum.tree_util) _list() (in module pybaum.registry_entries) _namedtuple() (in module pybaum.registry_entries)

_none() (in module pybaum.registry_entries) _numpy_array() (in module pybaum.registry_entries) _ordereddict() (in module pybaum.registry_entries) _pandas_dataframe() (in module pybaum.registry_entries) _pandas_series() (in module pybaum.registry_entries) _process_is_leaf() (in module pybaum.tree_util) _process_pytree_registry() (in module pybaum.tree_util) _tree_flatten() (in module pybaum.tree_util) _tree_unflatten() (in module pybaum.tree_util) _tree_yield() (in module pybaum.tree_util) _tuple() (in module pybaum.registry_entries) _unflatten_namedtuple() (in module pybaum.registry_entries) _unflatten_pandas_dataframe() (in module pybaum.registry_entries)

E

EQUALITY_CHECKERS (in module pybaum.equality)

F

FUNC_DICT (in module pybaum.registry_entries)

G

get_registry() (in module pybaum) (in module pybaum.registry)

get_type() (in module pybaum.typecheck)

I

IS_JAX_INSTALLED (in module pybaum.config)

IS_NUMPY_INSTALLED (in module pybaum.config) IS_PANDAS_INSTALLED (in module pybaum.config)

L

leaf_names() (in module pybaum) (in module pybaum.tree_util)

M

module pybaum pybaum._version pybaum.config pybaum.equality pybaum.registry pybaum.registry_entries pybaum.tree_util pybaum.typecheck

P

pybaum module pybaum._version module pybaum.config module pybaum.equality module

pybaum.registry module pybaum.registry_entries module pybaum.tree_util module pybaum.typecheck module

T

tree_equal() (in module pybaum) (in module pybaum.tree_util) tree_flatten() (in module pybaum) (in module pybaum.tree_util) tree_just_flatten() (in module pybaum) (in module pybaum.tree_util) tree_just_yield() (in module pybaum) (in module pybaum.tree_util) tree_map() (in module pybaum) (in module pybaum.tree_util)

tree_multimap() (in module pybaum) (in module pybaum.tree_util) tree_unflatten() (in module pybaum) (in module pybaum.tree_util) tree_update() (in module pybaum) (in module pybaum.tree_util) tree_yield() (in module pybaum) (in module pybaum.tree_util)