neuralqx.operators.computational.Euclidean4d.jax.area module

class AreaOperatorJax(H, edges, *, squared=False)

Bases: ComputationalJaxOperator

JAX-compatible (diagonal) area operator for a given surface S (list of edges). For σ[…, D] with D = n_edges_total * 3 (U(1)^3):

σp = σ[…, None, :] mels = (Σ_e ||m_e||_2) or (Σ_e ||m_e||_2^2) depending on squared.

property max_conn_size: int

The maximum number of non zero ⟨x|O|x’⟩ for every x.

property is_hermitian: bool

Returns true if this operator is hermitian.

property dtype

The dtype of the operator’s matrix elements ⟨σ|Ô|σ’⟩.

tree_flatten()

classmethod tree_unflatten(struct, leaves)