EvaluationOptions

class trainsum.options.EvaluationOptions(*, namespace, chunk_size=1024, optimizer='greedy')

Context manager for evalaute einsum options.

Parameters:

• namespace (ArrayNamespace)

• chunk_size (int)

• optimizer (Literal['optimal', 'dp', 'greedy', 'random-greedy', 'random-greedy-128', 'branch-all', 'branch-2', 'auto', 'auto-hq'])

optimizer: Literal['optimal', 'dp', 'greedy', 'random-greedy', 'random-greedy-128', 'branch-all', 'branch-2', 'auto', 'auto-hq']

Optimizer for the local einsum contraction paths.

chunk_size: int

Size of the chunks that are evaluated together.