neuralqx.operators.computational.Euclidean4d.jax package

ThiemannRegularisedVertexConstraintJax(lqx, vertex, *, apply_lapse=True, adjoint=False)
class VolumeOperatorJax(H, vertex)

Bases: ComputationalJaxOperator

JAX-compatible (diagonal) volume operator at a single vertex v for U(1)^3.

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.

class AreaDifferenceSquaredOperatorJax(H, edges)

Bases: ComputationalJaxOperator

Diagonal operator for the squared area difference between two edges e1,e2 in U(1)^3:

(||m_{e1}|| - ||m_{e2}||)^2 = ||m_{e1}||^2 + ||m_{e2}||^2 - 2 ||m_{e1}|| ||m_{e2}||.

class AreaDifferenceSquaredSurfacesOperatorJax(H, surfaces)

Bases: ComputationalJaxOperator

Diagonal operator for the squared area difference between two surfaces S1,S2 in U(1)^3:

(A(S1) - A(S2))^2, with A(S) = sum_{e in S} ||m_e||_2.

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.

Submodules