ExactOptions

class trainsum.options.ExactOptions(*, namespace, optimizer='greedy')

Context manager for exact einsum options.

Parameters:

• namespace (ArrayNamespace)

• 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.