Index

_

__init__() (spekk.spec.Spec method) (spekk.transformations.apply.Apply method) (spekk.transformations.axis.Axis method) (spekk.transformations.base.Buildable method) (spekk.transformations.base.PartialTransformation method) (spekk.transformations.base.Specced method) (spekk.transformations.base.Transformation method) (spekk.transformations.base.TransformedFunction method) (spekk.transformations.for_all.ForAll method) (spekk.transformations.reduce.Reduce method) (spekk.transformations.wrap.Wrap method) (spekk.trees.core.TraversalItem method) (spekk.trees.registry.DuckTypedTreeDef method) (spekk.trees.registry.TreeDef method) (spekk.trees.treelens.TreeLens method) (spekk.util.slicing.Sliceable method)

A

add_dimension() (spekk.spec.Spec method) Apply (class in spekk.transformations.apply) are_equal() (in module spekk.trees.core) args (spekk.transformations.apply.Apply attribute) (spekk.transformations.wrap.Wrap attribute)

at (spekk.spec.Spec property) (spekk.trees.treelens.TreeLens property) Axis (class in spekk.transformations.axis) AxisConcretizationError

B

becomes (spekk.transformations.axis.Axis attribute) build() (spekk.transformations.base.Buildable method) (spekk.transformations.base.Specced method) (spekk.transformations.base.TransformedFunction method)

Buildable (class in spekk.transformations.base)

C

compose() (in module spekk.transformations.common) concretize_axes() (in module spekk.transformations.axis) copy_with() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method)

create() (spekk.trees.registry.DuckTypedTreeDef method) (spekk.trees.registry.TreeDef method)

D

dimension (spekk.transformations.axis.Axis attribute) (spekk.transformations.reduce.Reduce attribute) dimensions (spekk.spec.Spec property)

dispatch_by_duck_type() (in module spekk.trees.registry) dispatch_by_type() (in module spekk.trees.registry) dispatch_treedef() (in module spekk.trees.registry) DuckTypedTreeDef (class in spekk.trees.registry)

E

enumerate (spekk.transformations.reduce.Reduce attribute)

extra_args (spekk.transformations.reduce.Reduce attribute) extra_kwargs (spekk.transformations.reduce.Reduce attribute)

F

f (spekk.transformations.apply.Apply attribute) (spekk.transformations.base.Specced attribute) (spekk.transformations.wrap.Wrap attribute)

filter() (in module spekk.trees.core) ForAll (class in spekk.transformations.for_all)

G

get() (in module spekk.trees.core) (spekk.spec.Spec method) (spekk.trees.registry.DuckTypedTreeDef method) (spekk.trees.registry.TreeDef method) (spekk.trees.treelens.TreeLens method)

get_args_for_index() (in module spekk.transformations.common) get_fn_name() (in module spekk.transformations.common) getitem_along_axis() (in module spekk.transformations.common)

H

has_dimension() (spekk.spec.Spec method) has_path() (in module spekk.trees.core)

has_subtree() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method) has_treedef() (in module spekk.trees.registry)

I

identity() (in module spekk.transformations.common) index_for() (spekk.spec.Spec method) initial_value (spekk.transformations.reduce.Reduce attribute) input_spec (spekk.transformations.base.Buildable attribute) (spekk.transformations.base.Specced attribute) (spekk.transformations.base.TransformedFunction attribute)

is_leaf (spekk.trees.core.TraversalItem attribute) is_leaf() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method)

K

keep (spekk.transformations.axis.Axis attribute) keys() (spekk.trees.registry.DuckTypedTreeDef method) (spekk.trees.registry.TreeDef method)

kwargs (spekk.transformations.apply.Apply attribute) (spekk.transformations.wrap.Wrap attribute)

L

leaves() (in module spekk.trees.core)

M

map_1_flattened() (in module spekk.transformations.common) merge() (in module spekk.trees.core) module spekk.spec spekk.transformations spekk.transformations.apply spekk.transformations.axis spekk.transformations.base spekk.transformations.common spekk.transformations.for_all spekk.transformations.reduce spekk.transformations.wrap spekk.trees spekk.trees.core spekk.trees.registry spekk.trees.treelens spekk.util spekk.util.flatten spekk.util.shape spekk.util.slicing spekk.util.validation

N

new_class() (spekk.trees.registry.DuckTypedTreeDef static method) (spekk.trees.registry.TreeDef static method)

new_dimensions() (spekk.transformations.axis.Axis method)

O

output_spec (spekk.transformations.base.Buildable attribute) (spekk.transformations.base.Specced attribute) (spekk.transformations.base.TransformedFunction attribute)

P

PartialTransformation (class in spekk.transformations.base) passed_spec (spekk.transformations.base.Buildable attribute) (spekk.transformations.base.Specced attribute) (spekk.transformations.base.TransformedFunction attribute)

path (spekk.trees.core.TraversalItem attribute) Product() (spekk.transformations.reduce.Reduce class method) prune_empty_branches() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method) python_vmap() (in module spekk.transformations.for_all)

R

Reduce (class in spekk.transformations.reduce) reduce_fn (spekk.transformations.reduce.Reduce attribute) reduce_impl (spekk.transformations.reduce.Reduce attribute) register_dispatch_fn() (in module spekk.trees.registry) register_type() (in module spekk.trees.registry) remove() (in module spekk.trees.core)

remove_dimension() (spekk.spec.Spec method) remove_subtree() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method) replace() (spekk.spec.Spec method) returned_spec (spekk.transformations.base.Buildable attribute) (spekk.transformations.base.Specced attribute) (spekk.transformations.base.TransformedFunction attribute)

S

set() (in module spekk.trees.core) (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method) size() (spekk.spec.Spec method) slice_array_1() (in module spekk.util.slicing) slice_data() (in module spekk.util.slicing) slice_spec() (in module spekk.util.slicing) Sliceable (class in spekk.util.slicing) Spec (class in spekk.spec) Specced (class in spekk.transformations.base) specced_map_reduce() (in module spekk.transformations.reduce) specced_vmap() (in module spekk.transformations.for_all) spekk.spec module spekk.transformations module spekk.transformations.apply module spekk.transformations.axis module spekk.transformations.base module spekk.transformations.common module

spekk.transformations.for_all module spekk.transformations.reduce module spekk.transformations.wrap module spekk.trees module spekk.trees.core module spekk.trees.registry module spekk.trees.treelens module spekk.util module spekk.util.flatten module spekk.util.shape module spekk.util.slicing module spekk.util.validation module Sum() (spekk.transformations.reduce.Reduce class method)

T

transform_function (spekk.transformations.base.Transformation attribute) transform_function() (spekk.transformations.apply.Apply method) (spekk.transformations.base.PartialTransformation method) (spekk.transformations.base.Transformation method) (spekk.transformations.for_all.ForAll method) (spekk.transformations.reduce.Reduce method) (spekk.transformations.wrap.Wrap method) transform_input_spec (spekk.transformations.base.Transformation attribute) transform_input_spec() (spekk.transformations.apply.Apply method) (spekk.transformations.base.PartialTransformation method) (spekk.transformations.base.Transformation method) (spekk.transformations.for_all.ForAll method) (spekk.transformations.reduce.Reduce method) (spekk.transformations.wrap.Wrap method) transform_output_spec (spekk.transformations.base.Transformation attribute) transform_output_spec() (spekk.transformations.apply.Apply method) (spekk.transformations.base.PartialTransformation method) (spekk.transformations.base.Transformation method) (spekk.transformations.for_all.ForAll method) (spekk.transformations.reduce.Reduce method) (spekk.transformations.wrap.Wrap method)

transform_spec (spekk.transformations.base.Specced attribute) Transformation (class in spekk.transformations.base) transformation (spekk.transformations.base.TransformedFunction attribute) TransformedFunction (class in spekk.transformations.base) TransformedFunctionError TraversalItem (class in spekk.trees.core) traverse() (in module spekk.trees.core) (spekk.transformations.base.TransformedFunction method) TreeDef (class in spekk.trees.registry) treedef() (in module spekk.trees.registry) TreeLens (class in spekk.trees.treelens)

U

update() (in module spekk.trees.core) update_leaves() (in module spekk.trees.core) (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method)

update_subtree() (spekk.spec.Spec method) (spekk.trees.treelens.TreeLens method)

V

validate() (in module spekk.util.validation) (spekk.spec.Spec method) ValidationError

value (spekk.trees.core.TraversalItem attribute) values() (spekk.trees.registry.DuckTypedTreeDef method) (spekk.trees.registry.TreeDef method)

W

Wrap (class in spekk.transformations.wrap)

wrapped_fn (spekk.transformations.base.TransformedFunction attribute)