The jax-array counterpart of OpTerm, registered as a pytree so that it can be
passed through jit-compiled functions.
-
__init__(opstr: str, strength: Array, indices: Array) → None
-
opstr: str
The operator string, treated as static auxiliary data.
-
strength: Array
A jax array of interaction strengths with shape (nterms,).
-
indices: Array
A jax array of site indices with shape (nterms, len(opstr)).
-
tree_flatten() → tuple[tuple[Array, Array], str]
Split into the array children (strength, indices) and static opstr.
-
classmethod tree_unflatten(aux_data: str, children: tuple[Array, Array]) → OpTermJAX
Rebuild the term from the static opstr and the array children.