neuralqx.operators.computational.misc.jax package

class U1HolonomyJax(H, edge_index, *, adjoint=False)

Bases: ComputationalJaxOperator

class NDU1HolonomyJax(H, edge_index, *, adjoint=False)

Bases: ComputationalJaxOperator

class MinimalLoopHolonomyJax(H, minimal_loop, *, copy=0, adjoint=False)

Bases: ComputationalJaxOperator

class NDMinimalLoopHolonomyJax(H, minimal_loop, *, adjoint=0)

Bases: ComputationalJaxOperator

class OneMinusHalfHolonomyJax(H, minimal_loop, *, copy=0)

Bases: ComputationalJaxOperator

class NumberJax(H, edge)

Bases: ComputationalJaxOperator

class IdentityJax(H, const=1.0)

Bases: ComputationalJaxOperator

Diagonal identity operator with optional scalar factor const (real or complex).

(IdentityJax(const=c)) |σ⟩ = c · |σ⟩

• Exactly one connection (diagonal).

• Hermitian iff c is real.

class ColoringJax(H, site, charge_vector)

Bases: ComputationalJaxOperator

JAX-native Coloring operator:

(ColoringJax at base site=e, charge_vector=(q1,…,q_g)) |σ⟩

= [σ[e + k*n_edges_total] == q_{k+1} ∀k=0..g-1] · |σ⟩

Works for generic gauge_dim (default use-case: U(1)^3).

class ChargeColoringJax(H, site, charge)

Bases: ComputationalJaxOperator

JAX-native ChargeColoring Operator:

(ChargeColoringJax at site=e, charge=q) |σ⟩ = [σ[e] == q] · |σ⟩