neuralqx.operators.computational.Euclidean4d.jax.gauss module¶

wrap_mod(vals_i32, delta_i32, state_min, mod_span, state_step)¶

Return type:: Array

gauss_constraint_kernel(sigma, comps_pad, signs_pad, *, gauge_dim, n_vertices, k_max, modular, mod_span, state_min, state_step)¶

class GaussConstraintOperatorJax(H, gauge_dim=None, *, modded=False)¶

Bases: ComputationalJaxOperator

JAX-compiled diagonal Gauß constraint operator:: sum_v || sum_{e->v} m_e - sum_{e<-v} m_e ||_2^2

Returns exactly one connection (the input σ) and its matrix element.

property is_hermitian: bool¶: Returns true if this operator is hermitian.

property dtype¶: The dtype of the operator’s matrix elements ⟨σ|Ô|σ’⟩.

property max_conn_size: int¶: The maximum number of non zero ⟨x|O|x’⟩ for every x.

tree_flatten()¶

classmethod tree_unflatten(aux, children)¶