Index

_

_canon_edge() (in module neuralqx.utils.symmetries.types) _collect_free_symbols() (in module neuralqx.experimental.operators.symbolic.ir.expressions) _collect_free_symbols_from_ops() (in module neuralqx.experimental.operators.symbolic.ir.update) _collect_free_symbols_pred() (in module neuralqx.experimental.operators.symbolic.ir.predicates) _is_reversed_vs_canon() (in module neuralqx.utils.symmetries.types)

_orient_like() (in module neuralqx.utils.symmetries.types) _render_amplitude() (in module neuralqx.experimental.operators.symbolic.ir.expressions) _render_predicate() (in module neuralqx.experimental.operators.symbolic.ir.predicates) _render_update_op() (in module neuralqx.experimental.operators.symbolic.ir.update) _scale_ir_term() (in module neuralqx.experimental.operators.symbolic.ir.term) _should_init_jax_distributed() (in module neuralqx.configs)

A

abort_requested (LiveMonitor property) AbstractBasisLayout (class in neuralqx.hilbert.utils.layout) AbstractGaugeGroup (class in neuralqx.gauge_groups), [1] (class in neuralqx.gauge_groups.abstract_gauge_group_interface) AbstractGraph (class in neuralqx.graph.core) (class in neuralqx.graph.core.abstract_graph) AbstractHilbertInterface (class in neuralqx.hilbert) (class in neuralqx.hilbert.abstract_hilbert_interface) AbstractHilbertSpace (class in neuralqx.hilbert) (class in neuralqx.hilbert.abstract_hilbert_core) AbstractLogger (class in neuralqx.utils.io.loggers) AbstractLqxInterface (class in neuralqx.lqx) (class in neuralqx.lqx.abstract_lqx_interface) AbstractLqxModel (class in neuralqx.lqx) (class in neuralqx.lqx.abstract_lqx_model) AbstractLqxSphericalBoSw (class in neuralqx.lqx.spherical.abstract_lqx_bosw_model) AbstractLqxSVQRLGModel (class in neuralqx.lqx.qr.abstract_single_vertex_qr_model) AbstractPrinter (class in neuralqx.utils.io.printing) AbstractSolver (class in neuralqx.solver) AbstractSymbolicArtifactStore (class in neuralqx.experimental.operators.symbolic.compiler.cache) (class in neuralqx.experimental.operators.symbolic.compiler.cache.store) AbstractSymbolicLowerer (class in neuralqx.experimental.operators.symbolic.compiler.lowering) (class in neuralqx.experimental.operators.symbolic.compiler.lowering.base) AbstractSymbolicOperator (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.core) (class in neuralqx.experimental.operators.symbolic.core.base) AbstractSymbolicPass (class in neuralqx.experimental.operators.symbolic.compiler.passes) (class in neuralqx.experimental.operators.symbolic.compiler.passes.base) AbstractVariationalDriver (class in neuralqx.driver.abstract_variational_driver) AcceptanceUnavailableError acting_on (LocalOperatorBase property) ActingOnModifier (class in neuralqx.utils.misc) AdaBelief (class in neuralqx.optimizer.types) Adagrad (class in neuralqx.optimizer.types) Adam (class in neuralqx.optimizer.types) add_attribute() (Triangulation method) add_field() (Logger method) adjacency (FreeEdgeFlipAllGauge attribute) (FreeEdgeFlipSingleGauge attribute) adjoint (ComputationalJaxOperator property) (ComputationalOperator property) (Euclidean3dVertexTRCAdjointOperator property) (Euclidean3dVertexTRCAdjointOperatorJax property) (Euclidean3dVertexTRCOperator property) (Euclidean3dVertexTRCOperatorJax property) (EuclideanConstraint property) (EuclideanConstraintJax property) (EuclideanVertexTRCAdjointOperator property) (EuclideanVertexTRCAdjointOperatorFast property) (EuclideanVertexTRCAdjointOperatorJax property) (EuclideanVertexTRCOperator property) (EuclideanVertexTRCOperatorFast property) (EuclideanVertexTRCOperatorJax property) (LorentzianConstraint property) (LorentzianConstraintJax property) (QRAnnihilation property) (QRAnnihilationJax property) (QRCreation property) (QRCreationJax property) affine() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3]

affine_penalty_cost_kernel() (in module neuralqx.vqs.mc.kernels) aggregate_run() (in module neuralqx.profile.aggregate) allgather() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) allowed_basis_states (AbstractHilbertSpace property) alpha (CosineDecay attribute) amplitude (EmissionSpec attribute), [1], [2], [3] (SymbolicIRTerm attribute), [1], [2], [3] AmplitudeExpr (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.expressions) AMSGrad (class in neuralqx.optimizer.types) annihilation() (AbstractLqxSVQRLGModel method) (in module neuralqx.operators.creation) (LqxSingleVertexQR method) (LqxSVQRLG method) area() (LqxWCL method) (LqxWCL1D method) (LqxWCL4D method) (LqxWclModel method) area_difference_squared_operator() (LqxWCL4D method) area_difference_squared_surfaces_operator() (LqxWCL4D method) AreaDifferenceEdgesError AreaDifferenceSquaredOperator (class in neuralqx.operators.computational.Euclidean4d.numba) (class in neuralqx.operators.computational.Euclidean4d.numba.area_difference) AreaDifferenceSquaredOperator() (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) AreaDifferenceSquaredOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax) (class in neuralqx.operators.computational.Euclidean4d.jax.area_difference) AreaDifferenceSquaredSurfacesOperator (class in neuralqx.operators.computational.Euclidean4d.numba) (class in neuralqx.operators.computational.Euclidean4d.numba.area_difference) AreaDifferenceSquaredSurfacesOperator() (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) AreaDifferenceSquaredSurfacesOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax) (class in neuralqx.operators.computational.Euclidean4d.jax.area_difference) AreaDifferenceSurfacesError AreaOperator (class in neuralqx.operators.computational.Euclidean4d.numba) (class in neuralqx.operators.computational.Euclidean4d.numba.area) AreaOperator() (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) AreaOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax) (class in neuralqx.operators.computational.Euclidean4d.jax.area) args (AmplitudeExpr attribute), [1], [2], [3] (PredicateExpr attribute), [1], [2], [3] array_complex() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) array_index() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) array_real() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) as_tuple() (NeuralqxVersion method) (Version method) Authenticator (class in neuralqx.utils.misc.auth) auto_serializable() (in module neuralqx.utils.serialization) AutoConstraintGaugeFixingConflictError Automorphism (class in neuralqx.utils.symmetries.types) (class in neuralqx.utils.symmetries.types.automorphism) automorphisms (GraphSymmetries property) AutoregressiveDirect (class in neuralqx.samplers.types) avmapr() (in module neuralqx.hilbert.constraints.utils)

B

b1 (AdaBelief attribute) (Adam attribute) (AMSGrad attribute) (Yogi attribute) b2 (AdaBelief attribute) (Adam attribute) (AMSGrad attribute) (Yogi attribute) backend (RuntimeInfo attribute), [1] (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] backend_preference (SymbolicCompilerOptions attribute), [1], [2], [3], [4] backend_target (SymbolicCompilationSignature attribute), [1], [2], [3], [4] barrier() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core)

batch_discrete_kernel() (in module neuralqx.vqs.mc.kernels) bcast() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) block_until_ready_tree() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.io) branch_tag (EmissionSpec attribute), [1], [2], [3] (SymbolicIRTerm attribute), [1], [2], [3] broadcast_from_canonical() (STMultiMCState method) build() (Optimizer method) (Sampler method) (Solvers method) build_lr() (in module neuralqx.optimizer.schedule) build_optimizer() (in module neuralqx.optimizer.builds) build_sampler() (in module neuralqx.samplers.builds) build_solver() (in module neuralqx.optimizer.solvers.builds)

C

c() (AbstractLqxSVQRLGModel method) (LqxSingleVertexQR method) (LqxSVQRLG method) cache_enabled (SymbolicCompilerOptions attribute), [1], [2], [3], [4] cache_key (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] cache_namespace (SymbolicCompilerOptions attribute), [1], [2], [3], [4] calculate_total_terms() (in module neuralqx.utils.parsing) canonical (NeuralqxVersion property) (Version property) canonical_jax_dtype() (in module neuralqx.utils.numbers) canonical_rotate() (in module neuralqx.graph.core.utils.build) canonical_state (STMultiMCState property) canonicalize_input() (in module neuralqx.operators.types.functional_local_operator.helpers) cast_MCState() (in module neuralqx.vqs.mc.mc_state.state) cast_operator_matrix_dtype() (in module neuralqx.operators.types.functional_local_operator.helpers) centered (RMSProp attribute) CFVTNet (class in neuralqx.nn.models) chain_length (MCState property) (MultiMCState property) (STMultiMCState property) characters (GroupProjectorDynamic attribute) charge_coloring() (in module neuralqx.operators) (in module neuralqx.operators.coloring) ChargeColoring (class in neuralqx.operators.computational.misc.numba) ChargeColoringJax (class in neuralqx.operators.computational.misc.jax) ChargeColoringOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) check_chunk_size() (in module neuralqx.vqs.mc.mc_state.state) check_distributed() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) check_hilbert() (in module neuralqx.vqs.mc) (in module neuralqx.vqs.mc.common) check_one_state_jax() (in module neuralqx.hilbert.utils) check_states() (ConstrainedHilbertU1Core method) check_valid_opmatrix() (in module neuralqx.operators.types.functional_local_operator.helpers) check_vertex_type_consistency() (in module neuralqx.graph.core.utils.checks) Cholesky (class in neuralqx.optimizer.solvers.types) choose_per_chain (MetropolisMultiHamiltonian attribute) (PTHamiltonianList attribute) chunk_size (MCState property) (MultiMCState property) CircularLadderGraph (class in neuralqx.graph) (class in neuralqx.graph.circular_ladder) class_ref() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.registry) coerce_amplitude_expr() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.ir) (in module neuralqx.experimental.operators.symbolic.ir.expressions) coerce_predicate_expr() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.ir) (in module neuralqx.experimental.operators.symbolic.ir.predicates) coerce_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.version) (in module neuralqx.utils.module.version_check) coerce_version_text() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.coercion) (in module neuralqx.utils.module.version_check) Coloring (class in neuralqx.operators.computational.misc.numba) coloring() (in module neuralqx.operators) (in module neuralqx.operators.coloring) ColoringJax (class in neuralqx.operators.computational.misc.jax)

ColoringOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) Common (class in neuralqx.samplers.types) commutator() (in module neuralqx.lqx.wcl.core.wcl_3d) compile_su2_graph_metadata() (in module neuralqx.experimental.hilbert.su2.metadata) compile_symbolic_operator() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.compiler) (in module neuralqx.experimental.operators.symbolic.compiler.compiler) compiled_operator (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] CompiledOperator (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.core) (class in neuralqx.experimental.operators.symbolic.core.compiled) complex (DTypePolicy attribute), [1] ComplexARNNDense (class in neuralqx.nn.models) ComputationalJaxOperator (class in neuralqx.operators.types) (class in neuralqx.operators.types.computational_operator) (class in neuralqx.operators.types.computational_operator.jax) ComputationalModelConcretizationWarning ComputationalOperator (class in neuralqx.operators.types) (class in neuralqx.operators.types.computational_operator) (class in neuralqx.operators.types.computational_operator.base) compute_chain_length() (in module neuralqx.vqs.mc.mc_state.state) compute_levi_civita_key() (in module neuralqx.utils.misc.levi_civita) compute_non_planar_graph_signs() (in module neuralqx.graph.core.utils.signs) compute_sgn_det() (in module neuralqx.graph.core.utils.signs) compute_sha256() (SerializerEngine static method) compute_sign_at_vertex() (in module neuralqx.graph.core.utils.signs) compute_signs() (in module neuralqx.graph.core.utils.signs) conjugate() (LocalOperatorBase method) conjugate_characters (GroupProjectorDynamic attribute) ConjugateGradient (class in neuralqx.optimizer.solvers.types) constant (LocalOperatorBase property) ConstrainedHilbertU1Core (class in neuralqx.hilbert.u1.constrained_core) constraint (AbstractGaugeGroup property) (AbstractLqxInterface property) (AbstractLqxModel property) (LqxSphericalBoSw property) (LqxSVQRLG property) (LqxWCL1D property) constraints_base (ConstrainedHilbertU1Core property) continue_simulation() (Solver method) convert_vertex_to_int() (GraphHandler method) copy() (LocalOperatorBase method) CosineDecay (class in neuralqx.optimizer.types) count_contributing_triplets_key() (in module neuralqx.utils.misc.graph) create_dressed_minimal_loop_key() (in module neuralqx.utils.misc.graph) create_dressed_nx_graph() (in module neuralqx.graph.core.utils.build) create_dual_nk_graph() (in module neuralqx.graph.core.utils.build) create_dual_nx_edges() (in module neuralqx.graph.core.utils.build) creation() (AbstractLqxSVQRLGModel method) (in module neuralqx.operators.creation) (LqxSingleVertexQR method) (LqxSVQRLG method) CrossProductInHigherDimensionsError curvature_constraint() (LqxWCL method) (LqxWCL1D method) (LqxWclModel method) cutoff (AbstractHilbertSpace property) Cycle (class in neuralqx.utils.symmetries.types) (class in neuralqx.utils.symmetries.types.cycle) cycles (GraphSymmetries property) CyclicGaugeFixingError

D

data (DeferredRuntimeLog property) dataclass() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) debug_flags (SymbolicCompilerOptions attribute), [1], [2], [3], [4] decay (RMSProp attribute) default_symbolic_artifact_store() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.compiler) (in module neuralqx.experimental.operators.symbolic.compiler.defaults) default_symbolic_lowerer_registry() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.compiler) (in module neuralqx.experimental.operators.symbolic.compiler.defaults) default_symbolic_pass_pipeline() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.compiler) (in module neuralqx.experimental.operators.symbolic.compiler.defaults) DefaultSolver (class in neuralqx.optimizer.solvers.types) defer_accumulation (DeferredRuntimeLog property) DeferredRuntimeLog (class in neuralqx.utils.io.runtime_loggers) (class in neuralqx.utils.io.runtime_loggers.deferred_runtime_log) delta (AbstractLqxSphericalBoSw property) (LqxBojowaldSwiderski property) (LqxSphericalBoSw property) DeniedExperimentalFeatureError DeniedExperimentalModuleImportError DependencyPolicy (class in neuralqx.utils.module.dependency_check) DependencyViolation deprecate_public_api() (in module neuralqx.utils.deprecation) deprecated() (in module neuralqx.utils.deprecation) deprecated_class() (in module neuralqx.utils.deprecation) deprecated_module() (in module neuralqx.utils.deprecation) deprecation_warning() (in module neuralqx.utils.deprecation) derive_output_shardings() (in module neuralqx.utils.jax.sharding) derived_fields() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) (Struct class method), [1] deserialize() (in module neuralqx.utils.serialization) (SerializerEngine static method) deserialize_MCState() (in module neuralqx.vqs.mc.mc_state.state) deserialize_MultiMCState() (in module neuralqx.solver) (in module neuralqx.solver.multi_state_solver) deserialize_STMultiMCState() (in module neuralqx.experimental.solver.stmh_solver), [1] detect_cpus_per_task() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) detect_edge_orientation_key() (in module neuralqx.utils.misc.graph) device_get_tree() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.io) dict_to_html() (in module neuralqx.utils.io.loggers.logger) dimensions (AbstractGaugeGroup property) (AbstractHilbertInterface property) (AbstractHilbertSpace property) dimensions_pretty (ConstrainedHilbertU1Core property) display_log() (Logger method) display_logs() (NQXPrinter method) DistributedRuntimeUnavailableWarning DistributedStateImportInSerialModeError DistributedStateImportMismatchError divide_by_group (GroupProjectorDynamic attribute)

DodecahedralGraph (class in neuralqx.graph) (class in neuralqx.graph.dodecahedral) DOperator (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.dsl) (class in neuralqx.experimental.operators.symbolic.dsl.op) draw_graph() (in module neuralqx.graph.core.utils.plotting) draw_nonplanar_graph() (in module neuralqx.graph.core.utils.plotting) dress_minimal_cycles_edges() (in module neuralqx.graph.core.utils.build) dressed_minimal_dual_loops() (AbstractGraph method) (Graph method) dressed_minimal_loops() (AbstractGraph method) (Graph method) dtype (AbstractHilbertSpace property) (AreaDifferenceSquaredOperator property) (AreaDifferenceSquaredOperatorJax property) (AreaDifferenceSquaredSurfacesOperator property) (AreaDifferenceSquaredSurfacesOperatorJax property) (AreaOperator property) (AreaOperatorJax property) (CompiledOperator attribute), [1], [2], [3] (ComputationalOperator property) (Euclidean3dVertexTRCOperator property) (Euclidean3dVertexTRCOperatorJax property) (EuclideanConstraint property) (EuclideanConstraintJax property) (EuclideanVertexTRCOperator property) (EuclideanVertexTRCOperatorFast property) (EuclideanVertexTRCOperatorJax property) (GaussConstraintOperator property) (GaussConstraintOperatorJax property) (LocalOperatorBase property) (LorentzianConstraint property) (LorentzianConstraintJax property) (QRAnnihilation property) (QRAnnihilationJax property) (QRCreation property) (QRCreationJax property) (QRFlux property) (QRFluxJax property) (SymbolicOperator attribute), [1], [2], [3] (VolumeOperator property), [1] (VolumeOperatorJax property), [1] dtype_name_map() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) dtype_str (SymbolicCompilationSignature attribute), [1], [2], [3], [4] (SymbolicOperatorIR attribute), [1], [2], [3] DTypePolicy (class in neuralqx.utils.dtypes) (class in neuralqx.utils.dtypes.policy) dual_nk_graph (AbstractGraph property) (Graph property) dual_nx_graph (AbstractGraph property) (Graph property) dump_recent_events() (in module neuralqx.debug) dump_stacks() (in module neuralqx.debug) DuplicateEdgesError duration_ms (SymbolicPassReport attribute), [1] dynamic_import_class() (SerializerEngine static method)

E

E() (AbstractLqxSVQRLGModel method) (LqxSingleVertexQR method) (LqxSVQRLG method) E_inv() (AbstractLqxSVQRLGModel method) (LqxSingleVertexQR method) (LqxSVQRLG method) edge_charges() (HilbertU1 method) edge_to_index() (AbstractGraph method) edge_to_site() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (UnconstrainedHilbertU1Core method) edges (AbstractGraph property) (GraphSymmetries property) edges_at() (AbstractGraph method) edges_from() (AbstractGraph method) edges_k (HalfLadderGraph property) edges_mu (HalfLadderGraph property) emissions (SymbolicIRTerm attribute), [1], [2], [3] EmissionSpec (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.term) emitted() (ExpressionContext method), [1], [2] (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.selectors) enable_fusion (SymbolicCompilerOptions attribute), [1], [2], [3], [4] enabled (NoVacuumProjector attribute) enabled() (in module neuralqx.profile) (Profiler method), [1] end_value (ExponentialDecay attribute) energy (VMC property) enforce_default_dependencies() (in module neuralqx.utils.module.dependency_check) enforce_policies() (in module neuralqx.utils.module.dependency_check) enforce_policy() (in module neuralqx.utils.module.dependency_check) ensure_batchdim() (in module neuralqx.hilbert.constraints.utils) eps (AdaBelief attribute) (Adagrad attribute) (Adam attribute) (AMSGrad attribute) (RMSProp attribute) (Yogi attribute) eps_root (AdaBelief attribute) (AMSGrad attribute) erase_parent_content() (Logger method) error (VerificationResult attribute) errors_only() (in module neuralqx.debug) escape_html() (in module neuralqx.utils.io.loggers.logger) estimate() (MultiStateVMC method) (SingleTrunkMultiHeadVMC method), [1] Euclidean3dVertexTRCAdjointOperator (class in neuralqx.operators.computational.Euclidean3d.numba.vertex_TRC) Euclidean3dVertexTRCAdjointOperatorJax (class in neuralqx.operators.computational.Euclidean3d.jax.vertex_TRC) Euclidean3dVertexTRCOperator (class in neuralqx.operators.computational.Euclidean3d.numba.vertex_TRC) Euclidean3dVertexTRCOperatorJax (class in neuralqx.operators.computational.Euclidean3d.jax.vertex_TRC) euclidean_constraint (AbstractLqxSVQRLGModel property) (LqxSingleVertexQR property) (LqxSVQRLG property)

EuclideanConstraint (class in neuralqx.operators.computational.qr.numba) (class in neuralqx.operators.computational.qr.numba.euclidean_constraint) EuclideanConstraintJax (class in neuralqx.operators.computational.qr.jax) (class in neuralqx.operators.computational.qr.jax.euclidean_constraint) EuclideanConstraintOperator() (in module neuralqx.operators.computational.qr) (in module neuralqx.operators.computational.qr.dispatch) EuclideanVertexTRCAdjointOperator (class in neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) EuclideanVertexTRCAdjointOperatorFast (class in neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) EuclideanVertexTRCAdjointOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax.vertex_TRC) EuclideanVertexTRCOperator (class in neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) EuclideanVertexTRCOperatorFast (class in neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) EuclideanVertexTRCOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax.vertex_TRC) event() (in module neuralqx.debug) Exact (class in neuralqx.samplers.types) exact_diagonalization() (AbstractLqxInterface method) ExactSampler (class in neuralqx.samplers.rules) ExceededDtypeValuesError expect() (in module neuralqx.vqs) (MCState method), [1], [2] (MultiMCState method), [1], [2] (MultiSolver method) (Solver method) (STMultiMCState method) (STMultiSolver method), [1] (VariationalState method), [1] expect_and_forces() (in module neuralqx.vqs) (MCState method), [1] (MultiMCState method), [1] (VariationalState method), [1] expect_and_grad() (in module neuralqx.vqs) (MCState method), [1] (MultiMCState method), [1] (VariationalState method), [1] expect_and_grad_biadjoint() (in module neuralqx.experimental.vqs.mc.mc_state.expect_grad_biadjoint) expect_and_grad_biadjoint_sequence() (in module neuralqx.experimental.vqs.mc.mc_state.expect_grad_biadjoint) ExpectationValueError experimental() (in module neuralqx.utils.experimental) ExponentialDecay (class in neuralqx.optimizer.types) export() (AbstractGraph method) (GraphHandler method) (in module neuralqx.experimental.utils.struct.io) export_info() (in module neuralqx.optimizer.export) (in module neuralqx.optimizer.solvers.export) (in module neuralqx.samplers.export) (Optimizer method) (Sampler method) (Solvers method) export_state() (MultiSolver method) (Solver method) (STMultiSolver method), [1] ExpressionContext (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.dsl) (class in neuralqx.experimental.operators.symbolic.dsl.context) extract_from_html() (Authenticator static method)

F

factorial() (in module neuralqx.utils.misc.arithmetic) fanout_hint (SymbolicIRTerm attribute), [1], [2], [3] fidelity_expect_and_grad_joint() (in module neuralqx.driver.multi_state_vmc) fidelity_expect_and_grad_joint_tied() (in module neuralqx.experimental.driver.stmh_multi_vmc), [1] fidelity_matrix() (MultiMCState method) field() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.fields) FieldKind (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.fields) fields() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) (Struct class method), [1] FieldSpec (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.fields) find_available_port() (LiveMonitor static method) find_connected_edges() (in module neuralqx.graph.core.utils.build) find_corner_vertex_keyed() (in module neuralqx.utils.misc.graph) find_free_port() (in module neuralqx.utils.jax.djaxrun) find_minimal_loops_with_edges_key() (in module neuralqx.utils.misc.graph) flatten() (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (UnconstrainedHilbertU1Core method) flip_state() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (in module neuralqx.hilbert.u1.operations.flip) (UnconstrainedHilbertU1Core method)

flip_state_all_gd() (ConstrainedHilbertU1Core method) flush() (DeferredRuntimeLog method) (in module neuralqx.profile) (Profiler method), [1] flux() (in module neuralqx.operators.creation) (LqxBojowaldSwiderski method) force_to_grad() (in module neuralqx.vqs.mc.common) forces_expect_hermitian() (in module neuralqx.vqs.mc.mc_state.expect_forces) forces_expect_hermitian_chunked() (in module neuralqx.vqs.mc.mc_state.expect_forces_chunked) forces_expect_hermitian_sequence() (in module neuralqx.vqs.mc.mc_state.expect_forces) forces_expect_hermitian_sequence_chunked() (in module neuralqx.vqs.mc.mc_state.expect_forces_chunked) format (VerificationResult attribute) format_value() (in module neuralqx.utils.io.loggers.logger) FreeEdgeFlipAllGauge (class in neuralqx.hilbert.u1.operations.moves) FreeEdgeFlipSingleGauge (class in neuralqx.hilbert.u1.operations.moves) from_state_dict() (in module neuralqx.experimental.utils.struct.io) FrozenStructError full_complex() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) full_real() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) FunctionalLocalOperator (class in neuralqx.operators.types) (class in neuralqx.operators.types.functional_local_operator.numba) FunctionalLocalOperatorJax (class in neuralqx.operators.types.functional_local_operator.jax)

G

gauge_dimensions (AbstractHilbertInterface property) (AbstractHilbertSpace property) (AbstractLqxInterface property) (AbstractLqxModel property) gauge_fixing (ConstrainedHilbertU1Core property) gauge_group (AbstractLqxInterface property) (AbstractLqxModel property) GaugeCoord (class in neuralqx.hilbert.u1.layout) (class in neuralqx.hilbert.u1.layout.strided) GaugeFixingTopo (class in neuralqx.hilbert.u1.constrained_core) GaugeLayout (class in neuralqx.hilbert.u1.layout) (class in neuralqx.hilbert.u1.layout.strided) gauss_constraint_kernel() (in module neuralqx.operators.computational.Euclidean4d.jax.gauss) GaussConstraintOperator (class in neuralqx.operators.computational.Euclidean4d.numba) (class in neuralqx.operators.computational.Euclidean4d.numba.gauss) GaussConstraintOperator() (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) GaussConstraintOperatorJax (class in neuralqx.operators.computational.Euclidean4d.jax) (class in neuralqx.operators.computational.Euclidean4d.jax.gauss) generate_charge_vectors() (in module neuralqx.utils.misc.charges) generate_constraint_array() (in module neuralqx.hilbert.constraints.utils) generate_key_pair() (Authenticator static method) generate_plus_minus_one() (in module neuralqx.utils.misc.arithmetic) get_abstract_mesh() (in module neuralqx.utils.jax.sharding) get_attr_val() (in module neuralqx.utils.parsing) get_common_base_vertex() (in module neuralqx.utils.misc.graph) get_conn_filtered() (FunctionalLocalOperator method) get_conn_flattened() (ComputationalOperator method) (FunctionalLocalOperator method) get_conn_padded() (ComputationalJaxOperator method) (FunctionalLocalOperatorJax method) get_distributed_info_dict() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) get_dtype_policy() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) get_edges_at_base_vertex() (in module neuralqx.utils.misc.graph) get_edges_at_k_vertex() (HalfLadderGraph method) get_fully_qualified_name() (SerializerEngine static method) get_hash() (in module neuralqx.utils.misc.auth) get_ijk_key() (in module neuralqx.utils.misc.levi_civita) get_kminus_kplus() (HalfLadderGraph method) get_local_kernel() (in module neuralqx.vqs) (in module neuralqx.vqs.mc) get_local_kernel_arguments() (in module neuralqx.vqs) (in module neuralqx.vqs.mc) get_log() (Logger method) get_logfile() (in module neuralqx.debug) get_ml_graph_shift_key() (in module neuralqx.utils.misc.graph)

get_modifier_string() (in module neuralqx.utils.misc.modifiers) get_modifier_value() (in module neuralqx.utils.misc.modifiers) get_module_neuralqx_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.module_version) (in module neuralqx.utils.module.version_check) get_module_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.module_version) (in module neuralqx.utils.module.version_check) get_module_version_string() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.module_version) (in module neuralqx.utils.module.version_check) get_profiler() (in module neuralqx.profile) (in module neuralqx.profile.profiler) get_quantum_number() (in module neuralqx.operators) (in module neuralqx.operators.number) get_reversed_keyed_edge() (in module neuralqx.utils.misc.graph) get_serializable_state() (SerializerEngine static method) get_sgn() (in module neuralqx.utils.misc.arithmetic) get_sign_of_tuple_key() (in module neuralqx.utils.misc.levi_civita) get_signed_value() (in module neuralqx.utils.misc.arithmetic) get_smallest_dual_loops() (GraphHandler method) get_smallest_loops() (GraphHandler method) get_stats_mean() (in module neuralqx.experimental.vqs.mc.mc_state.utils) get_triplets_key() (in module neuralqx.utils.misc.graph) get_true_edge_triplet() (in module neuralqx.utils.misc.graph) get_value_of() (Logger method) get_vertices_from_edges() (in module neuralqx.graph.core.utils.checks) get_vertices_from_minimal_loop() (in module neuralqx.utils.misc.graph) grad() (VariationalState method) GradientGradientCosineCallback (class in neuralqx.callbacks) (class in neuralqx.callbacks.pairwise_gradient_cosine_callback) graph (AbstractGaugeGroup property) (AbstractHilbertInterface property) (AbstractHilbertSpace property) (AbstractLqxInterface property) (AbstractLqxModel property) Graph (class in neuralqx.graph) (class in neuralqx.graph.graph) graph_symmetries (GraphSymmetries property) GraphHandler (class in neuralqx.graph.core.graph_handler) GraphSymmetries (class in neuralqx.utils.symmetries) (class in neuralqx.utils.symmetries.graph_symmetries) GraphUnavailableWarning Grid2D (class in neuralqx.graph) (class in neuralqx.graph.grid_2d) ground_energy (AbstractLqxInterface property) ground_state (AbstractLqxInterface property) GroupProjectorDynamic (class in neuralqx.nn.projectors.group_projector) (class in neuralqx.nn.projectors.group_projector.group_projector)

H

H (LqxWCL4D property) HalfLadderGraph (class in neuralqx.graph) (class in neuralqx.graph.half_ladder_graph) hamiltonian_boundaries() (LqxSphericalBoSw method) hamiltonian_list() (LqxSphericalBoSw method) has_ground_energy (AbstractLqxInterface property) has_ground_state (AbstractLqxInterface property) has_k_vertex_neighbours() (HalfLadderGraph method) hermitian_flag_for_sum_with_scalar() (in module neuralqx.utils.numbers) HexagonalLatticeGraph (class in neuralqx.graph) (class in neuralqx.graph.hexagonal_lattice) hilbert (AbstractGaugeGroup property) (AbstractHilbertInterface property) (AbstractHilbertSpace property) (AbstractLqxInterface property) (AbstractLqxModel property) (ConstrainedHilbertU1Core property) (MultiMCState property) (STMultiMCState property) (SymbolicOperator attribute), [1], [2], [3] (VariationalState property)

hilbert_dimensions (AbstractLqxInterface property) (AbstractLqxModel property) hilbert_netket (AbstractHilbertInterface property) hilbert_size (SymbolicCompilationSignature attribute), [1], [2], [3], [4] (SymbolicOperatorIR attribute), [1], [2], [3] HilbertSpaceGaugeGroupDimensionsMismatchError HilbertStateEnumerator (class in neuralqx.hilbert.utils.index) HilbertU1 (class in neuralqx.hilbert.u1) (class in neuralqx.hilbert.u1.interface) holonomy() (AbstractLqxSphericalBoSw method) (in module neuralqx.operators) (in module neuralqx.operators.holonomies) (LqxBojowaldSwiderski method) (LqxSphericalBoSw method) (LqxWCL method) (LqxWCL1D method) (LqxWCL3D method) (LqxWclModel method) HypercubeGraph (class in neuralqx.graph) (class in neuralqx.graph.hypercube)

I

Identity (class in neuralqx.operators.computational.misc.numba) identity() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) IdentityJax (class in neuralqx.operators.computational.misc.jax) IdentityOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) ignore_chunk_warning() (in module neuralqx.vqs.mc.mc_state.expect_grad_chunked) Immirzi (AbstractLqxSphericalBoSw property) (LqxBojowaldSwiderski property) (LqxSphericalBoSw property) import_optional_dep() (in module neuralqx.utils.module.optional_dependency) import_state() (MultiSolver method) (Solver method) (STMultiSolver method), [1] IncompatibleHilbertSpaceError IncompatibleJaxOperatorError IncompatibleModdedOperatorWarning IncompatibleNonGIOperatorError IncompatibleNonPlanarGraphModel IncompatiblePlanarGraphModel IncorrectEdgeFormatError IncorrectGaugeFixingArrayError IncorrectMonitoringValueError index (AbstractHilbertSpace property) (DTypePolicy attribute), [1] index_sets (KBodyIteratorSpec attribute), [1], [2], [3] index_to_edge() (AbstractGraph method) index_to_label() (in module neuralqx.graph.core.utils.build) init() (MCState method) init_constraint() (AbstractGaugeGroup method) (U1GaugeGroup method) init_constraint_terms() (LqxWCL4D method) init_node_connectivity() (in module neuralqx.graph.core.utils.build) init_parameters() (MultiMCState method) (VariationalState method) initial_accumulator_value (Adagrad attribute) initialise() (in module neuralqx.debug) initialize_constraint() (AbstractLqxInterface method) (LqxBojowaldSwiderski method) (LqxSingleVertexQR method) initialize_vmc() (MultiSolver method) (Solver method) (STMultiSolver method), [1] InMemorySymbolicArtifactStore (class in neuralqx.experimental.operators.symbolic.compiler.cache) (class in neuralqx.experimental.operators.symbolic.compiler.cache.store) inner_vertices (AbstractLqxSphericalBoSw property) (LqxBojowaldSwiderski property) (LqxSphericalBoSw property) InvalidChargeError InvalidCutoffError InvalidEdgeSelectionError InvalidFreeEdgeSelectionError InvalidIndexError InvalidMelsFunctionError InvalidOperatorsInSequenceError InvalidOperatorsSequenceError InvalidSU2CutoffError InvalidSU2StateLayoutError InvalidSurfaceError InvalidVersion, [1] inverse_mapping() (GraphHandler method)

InverseExpectationCost (class in neuralqx.operators) io_trace() (in module neuralqx.debug) irrep_dim (GroupProjectorDynamic attribute) is_abelian (AbstractGaugeGroup property) (U1GaugeGroup property) is_computational (AbstractGaugeGroup property) (AbstractLqxModel property) is_connected (HalfLadderGraph property) is_distributed_array() (in module neuralqx.utils.jax.sharding) is_enabled() (in module neuralqx.debug) is_finite (AbstractHilbertInterface property) is_gauge_invariant (AbstractHilbertInterface property) is_gauge_invariant() (ConstrainedHilbertU1Core method) is_global_master() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) is_group_averaged (MultiMCState property) is_hermitian (AreaDifferenceSquaredOperator property) (AreaDifferenceSquaredOperatorJax property) (AreaDifferenceSquaredSurfacesOperator property) (AreaDifferenceSquaredSurfacesOperatorJax property) (AreaOperator property) (AreaOperatorJax property) (CompiledOperator attribute), [1], [2], [3] (ComputationalOperator property) (Euclidean3dVertexTRCOperator property) (Euclidean3dVertexTRCOperatorJax property) (EuclideanConstraint property) (EuclideanConstraintJax property) (EuclideanVertexTRCOperator property) (EuclideanVertexTRCOperatorFast property) (EuclideanVertexTRCOperatorJax property) (GaussConstraintOperator property) (GaussConstraintOperatorJax property) (LocalOperatorBase property) (LorentzianConstraint property) (LorentzianConstraintJax property) (QRAnnihilation property) (QRAnnihilationJax property) (QRCreation property) (QRCreationJax property) (QRFlux property) (QRFluxJax property) (SymbolicOperator attribute), [1], [2], [3] (SymbolicOperatorIR attribute), [1], [2], [3] (VolumeOperator property), [1] (VolumeOperatorJax property), [1] is_hermitian() (in module neuralqx.operators.types.functional_local_operator.base) is_indexable (AbstractHilbertInterface property) is_jax (AbstractGaugeGroup property) (AbstractLqxSVQRLGModel property) is_ket_action (ComputationalJaxOperator property) (ComputationalOperator property) is_master() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) is_planar (AbstractGraph property) is_registered_pytree_type() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.pytree) is_running (LiveMonitor property) is_scalar_like() (in module neuralqx.utils.numbers) is_sharded_array() (in module neuralqx.utils.jax.sharding) iter() (AbstractVariationalDriver method) iterator (SymbolicIRTerm attribute), [1], [2], [3]

J

jax_complex_dtype() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) jax_index_dtype() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime)

jax_real_dtype() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) JAXSymbolicLowerer (class in neuralqx.experimental.operators.symbolic.compiler.lowering) (class in neuralqx.experimental.operators.symbolic.compiler.lowering.jax_lowerer) jit_evaluate() (in module neuralqx.vqs.mc.mc_state.state)

K

K5Graph (class in neuralqx.graph) (class in neuralqx.graph.k5_graph) KBodyIteratorSpec (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.term)

kind (UpdateOp attribute), [1], [2], [3]

L

labels (KBodyIteratorSpec attribute), [1], [2], [3] ladder_length (CircularLadderGraph property) (LadderGraph property) LadderGraph (class in neuralqx.graph) (class in neuralqx.graph.ladder_graph) lambda_ortho (MultiSolver property) (STMultiSolver property), [1] layout (AbstractHilbertSpace property) (ConstrainedHilbertU1Core property) (UnconstrainedHilbertU1Core property) LinearDecay (class in neuralqx.optimizer.types) LiveMonitor (class in neuralqx.callbacks.live_monitoring) LiveMonitoringCallback (class in neuralqx.callbacks) (class in neuralqx.callbacks.live_monitoring) LiveMonitoringUnavailableWarning load() (in module neuralqx.experimental.utils.struct.io) load_from_file() (in module neuralqx.utils.serialization) (SerializerEngine static method) load_multi_partial() (in module neuralqx.utils.serialization) local_estimators() (in module neuralqx.vqs.mc.mc_state.state) (MCState method) (MultiMCState method) local_operators_to_pauli_strings() (in module neuralqx.operators.types.functional_local_operator.convert) local_rank (RuntimeInfo attribute), [1] local_size (AbstractHilbertInterface property) (AbstractHilbertSpace property) local_value_kernel() (in module neuralqx.vqs.mc.kernels) local_value_kernel_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_conn_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_conn_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action_conn_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action_conn_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_jax_ket_action_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_ket_action() (in module neuralqx.vqs.mc.kernels) local_value_kernel_ket_action_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_ket_action_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_ket_action_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_penalty_cost() (in module neuralqx.vqs.mc.kernels) local_value_kernel_penalty_cost_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_penalty_cost_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_penalty_cost_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_conn_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_conn_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_ket_action() (in module neuralqx.vqs.mc.kernels)

local_value_kernel_squared_jax_ket_action_chunked() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_ket_action_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_squared_jax_ket_action_factored() (in module neuralqx.vqs.mc.kernels) local_value_kernel_variance() (in module neuralqx.vqs.mc.kernels) local_value_op_op_cost() (in module neuralqx.vqs.mc.kernels) local_value_op_op_cost_chunked() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_chunked() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_factored() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_ket_action() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_ket_action_chunked() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_ket_action_chunked_factored() (in module neuralqx.vqs.mc.kernels) local_value_squared_kernel_ket_action_factored() (in module neuralqx.vqs.mc.kernels) LocalGraphNQS (class in neuralqx.nn.models) LocalOperatorBase (class in neuralqx.operators.types.functional_local_operator.base) log() (LiveMonitor method) (Logger method) (PrintHandler method) log_module_attributes() (in module neuralqx.utils.parsing) log_once() (in module neuralqx.debug) log_value() (MCState method) (MultiMCState method) Logger (class in neuralqx.utils.io.loggers) (class in neuralqx.utils.io.loggers.logger) lorentzian_constraint (AbstractLqxSVQRLGModel property) (LqxSingleVertexQR property) (LqxSVQRLG property) LorentzianConstraint (class in neuralqx.operators.computational.qr.numba) (class in neuralqx.operators.computational.qr.numba.lorentzian_constraint) LorentzianConstraintJax (class in neuralqx.operators.computational.qr.jax) (class in neuralqx.operators.computational.qr.jax.lorentzian_constraint) LorentzianConstraintOperator() (in module neuralqx.operators.computational.qr) (in module neuralqx.operators.computational.qr.dispatch) lowerer_name (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] LqxBojowaldSwiderski (class in neuralqx.lqx) (class in neuralqx.lqx.spherical) (class in neuralqx.lqx.spherical.spherical_bojowald_swiderski_interface) LqxSingleVertexQR (class in neuralqx.lqx) (class in neuralqx.lqx.qr) (class in neuralqx.lqx.qr.single_vertex_qr_interface) LqxSphericalBoSw (class in neuralqx.lqx.spherical.core) (class in neuralqx.lqx.spherical.core.spherical_bosw) LqxSVQRLG (class in neuralqx.lqx.qr.core) (class in neuralqx.lqx.qr.core.single_vertex_qrlg) LqxWCL (class in neuralqx.lqx) (class in neuralqx.lqx.wcl) (class in neuralqx.lqx.wcl.euclidean_interface) LqxWCL1D (class in neuralqx.lqx.wcl.core) (class in neuralqx.lqx.wcl.core.wcl_1d) LqxWCL3D (class in neuralqx.lqx.wcl.core) (class in neuralqx.lqx.wcl.core.wcl_3d) LqxWCL4D (class in neuralqx.lqx.wcl.core) (class in neuralqx.lqx.wcl.core.wcl_4d) LqxWclModel (class in neuralqx.lqx.wcl) (class in neuralqx.lqx.wcl.abstract_lqx_wcl_model) LU (class in neuralqx.optimizer.solvers.types)

M

m (Grid2D property) (HexagonalLatticeGraph property) (TriangularLatticeGraph property) main() (in module neuralqx.utils.jax.djaxrun) major (NeuralqxVersion property) (Version property) make_cfvtnet_kwargs() (in module neuralqx.nn.models) make_grad_qgt_compatible() (in module neuralqx.experimental.vqs.mc.mc_state.utils) make_hashable() (in module neuralqx.hilbert.utils) make_learning_rate() (in module neuralqx.optimizer.schedule) make_local_graph_nqs_kwargs() (in module neuralqx.nn.models) make_permutation_from_order() (GraphSymmetries method) make_shared_stmh_state() (in module neuralqx.experimental.vqs.mc.mc_state.stmh_state) make_stmh_head_models() (in module neuralqx.experimental.nn.projectors.stmh) map_distinct_vertices() (in module neuralqx.graph.core.utils.build) map_edge() (Symmetry method) map_edge_string() (in module neuralqx.hilbert.constraints.utils) map_edge_to_dual_vertex() (GraphHandler method) map_edges() (in module neuralqx.graph.core.utils.build) map_vertex() (Automorphism method) mapping (AbstractGraph property) max_conn_size (AreaDifferenceSquaredOperatorJax property) (AreaDifferenceSquaredSurfacesOperatorJax property) (AreaOperatorJax property) (Euclidean3dVertexTRCOperator property) (Euclidean3dVertexTRCOperatorJax property) (EuclideanConstraintJax property) (EuclideanVertexTRCOperatorJax property) (GaussConstraintOperatorJax property) (LocalOperatorBase property) (LorentzianConstraintJax property) (QRAnnihilationJax property) (QRCreationJax property) (QRFluxJax property) (VolumeOperatorJax property), [1] max_nonzero_per_row() (in module neuralqx.operators.types.functional_local_operator.compile_helpers) maximum (DependencyPolicy attribute) maybe_post_load_hook() (SerializerEngine static method) maybe_sync() (Profiler method) MCState (class in neuralqx.vqs) (class in neuralqx.vqs.mc) (class in neuralqx.vqs.mc.mc_state) (class in neuralqx.vqs.mc.mc_state.state) mel_cutoff (LocalOperatorBase property) metadata (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] (SymbolicIRTerm attribute), [1], [2], [3] (SymbolicOperatorIR attribute), [1], [2], [3] (SymbolicPassReport attribute), [1] MetropolisExchange (class in neuralqx.samplers.types) MetropolisHamiltonian (class in neuralqx.samplers.types) MetropolisKLocal (class in neuralqx.samplers.types) MetropolisKLocalRule (class in neuralqx.samplers) (class in neuralqx.samplers.rules) MetropolisLocal (class in neuralqx.samplers.types) MetropolisMultiHamiltonian (class in neuralqx.samplers.types) minimal_dual_loops() (AbstractGraph method) (Graph method) minimal_loop_holonomy() (LqxWCL method) (LqxWCL1D method) (LqxWCL3D method) (LqxWclModel method) minimal_loops() (AbstractGraph method) (Graph method) MinimalLoopHolonomy (class in neuralqx.operators.computational.misc.numba) MinimalLoopHolonomyJax (class in neuralqx.operators.computational.misc.jax) MinimalLoopHolonomyOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) minimum (DependencyPolicy attribute) minor (NeuralqxVersion property) (Version property) minus_key() (in module neuralqx.utils.misc.arithmetic) MissingPenaltyFactorError mod_add() (in module neuralqx.utils.misc.arithmetic) mod_add_jax() (in module neuralqx.utils.jit) mod_add_njit() (in module neuralqx.utils.jit) mod_sum() (in module neuralqx.utils.misc.arithmetic) mod_sum_jax() (in module neuralqx.utils.jit) mode (SymbolicOperatorIR attribute), [1], [2], [3] model (AbstractLqxInterface property) (MCState property) (MultiMCState property) (STMultiMCState property) model_name (AbstractLqxInterface property) (AbstractLqxModel property) model_state (MultiMCState property) (VariationalState property) module neuralqx neuralqx.callbacks neuralqx.callbacks.converged_state_export neuralqx.callbacks.live_monitoring neuralqx.callbacks.pairwise_gradient_cosine_callback neuralqx.configs neuralqx.debug neuralqx.driver neuralqx.driver.abstract_variational_driver neuralqx.driver.multi_state_vmc neuralqx.driver.vmc neuralqx.experimental neuralqx.experimental.driver, [1] neuralqx.experimental.driver.stmh_multi_vmc, [1] neuralqx.experimental.hilbert neuralqx.experimental.hilbert.su2 neuralqx.experimental.hilbert.su2.index neuralqx.experimental.hilbert.su2.layout neuralqx.experimental.hilbert.su2.metadata neuralqx.experimental.hilbert.su2.operations neuralqx.experimental.nn neuralqx.experimental.nn.projectors neuralqx.experimental.nn.projectors.stmh neuralqx.experimental.operators, [1] neuralqx.experimental.operators.symbolic, [1] neuralqx.experimental.operators.symbolic.compiler neuralqx.experimental.operators.symbolic.compiler.cache neuralqx.experimental.operators.symbolic.compiler.cache.store neuralqx.experimental.operators.symbolic.compiler.compiler neuralqx.experimental.operators.symbolic.compiler.core neuralqx.experimental.operators.symbolic.compiler.core.artifact neuralqx.experimental.operators.symbolic.compiler.core.context neuralqx.experimental.operators.symbolic.compiler.core.options neuralqx.experimental.operators.symbolic.compiler.core.pass_report neuralqx.experimental.operators.symbolic.compiler.core.pipeline neuralqx.experimental.operators.symbolic.compiler.core.signature neuralqx.experimental.operators.symbolic.compiler.defaults neuralqx.experimental.operators.symbolic.compiler.lowering neuralqx.experimental.operators.symbolic.compiler.lowering.base neuralqx.experimental.operators.symbolic.compiler.lowering.jax_lowerer neuralqx.experimental.operators.symbolic.compiler.lowering.registry neuralqx.experimental.operators.symbolic.compiler.passes neuralqx.experimental.operators.symbolic.compiler.passes.analysis neuralqx.experimental.operators.symbolic.compiler.passes.base neuralqx.experimental.operators.symbolic.compiler.passes.fusion neuralqx.experimental.operators.symbolic.compiler.passes.normalization neuralqx.experimental.operators.symbolic.compiler.passes.validation neuralqx.experimental.operators.symbolic.core neuralqx.experimental.operators.symbolic.core.base neuralqx.experimental.operators.symbolic.core.compiled neuralqx.experimental.operators.symbolic.core.operator neuralqx.experimental.operators.symbolic.core.sum neuralqx.experimental.operators.symbolic.dsl neuralqx.experimental.operators.symbolic.dsl.context neuralqx.experimental.operators.symbolic.dsl.op neuralqx.experimental.operators.symbolic.dsl.rewrite neuralqx.experimental.operators.symbolic.dsl.selectors neuralqx.experimental.operators.symbolic.ir neuralqx.experimental.operators.symbolic.ir.expressions neuralqx.experimental.operators.symbolic.ir.predicates neuralqx.experimental.operators.symbolic.ir.program neuralqx.experimental.operators.symbolic.ir.term neuralqx.experimental.operators.symbolic.ir.update neuralqx.experimental.operators.symbolic.ir.validate neuralqx.experimental.solver, [1] neuralqx.experimental.solver.stmh_solver, [1] neuralqx.experimental.utils neuralqx.experimental.utils.errors neuralqx.experimental.utils.struct neuralqx.experimental.utils.struct.base neuralqx.experimental.utils.struct.fields neuralqx.experimental.utils.struct.io neuralqx.experimental.utils.struct.pytree neuralqx.experimental.utils.struct.registry neuralqx.experimental.vqs, [1] neuralqx.experimental.vqs.mc, [1] neuralqx.experimental.vqs.mc.mc_state neuralqx.experimental.vqs.mc.mc_state.expect_grad_biadjoint neuralqx.experimental.vqs.mc.mc_state.stmh_state neuralqx.experimental.vqs.mc.mc_state.utils neuralqx.gauge_groups neuralqx.gauge_groups.abstract_gauge_group_interface neuralqx.gauge_groups.u1 neuralqx.gauge_groups.u1.u1_gauge_group neuralqx.graph neuralqx.graph.circular_ladder neuralqx.graph.core neuralqx.graph.core.abstract_graph neuralqx.graph.core.graph_handler neuralqx.graph.core.utils neuralqx.graph.core.utils.build neuralqx.graph.core.utils.checks neuralqx.graph.core.utils.embedding neuralqx.graph.core.utils.plotting neuralqx.graph.core.utils.sign_utils neuralqx.graph.core.utils.signs neuralqx.graph.dodecahedral neuralqx.graph.graph neuralqx.graph.grid_2d neuralqx.graph.half_ladder_graph neuralqx.graph.hexagonal_lattice neuralqx.graph.hypercube neuralqx.graph.k5_graph neuralqx.graph.ladder_graph neuralqx.graph.single_vertex neuralqx.graph.triangular_lattice neuralqx.hilbert neuralqx.hilbert.abstract_hilbert_core neuralqx.hilbert.abstract_hilbert_interface neuralqx.hilbert.constraints neuralqx.hilbert.constraints.u1_constraint neuralqx.hilbert.constraints.utils neuralqx.hilbert.u1 neuralqx.hilbert.u1.constrained_core neuralqx.hilbert.u1.index neuralqx.hilbert.u1.index.enumerator neuralqx.hilbert.u1.index.mapping neuralqx.hilbert.u1.index.utils neuralqx.hilbert.u1.interface neuralqx.hilbert.u1.layout neuralqx.hilbert.u1.layout.strided neuralqx.hilbert.u1.operations neuralqx.hilbert.u1.operations.flip neuralqx.hilbert.u1.operations.moves neuralqx.hilbert.u1.operations.random neuralqx.hilbert.u1.unconstrained_core neuralqx.hilbert.utils neuralqx.hilbert.utils.index neuralqx.hilbert.utils.index.dispatch neuralqx.hilbert.utils.layout neuralqx.lqx neuralqx.lqx.abstract_lqx_interface neuralqx.lqx.abstract_lqx_model neuralqx.lqx.qr neuralqx.lqx.qr.abstract_single_vertex_qr_model neuralqx.lqx.qr.core neuralqx.lqx.qr.core.single_vertex_qrlg neuralqx.lqx.qr.single_vertex_qr_interface neuralqx.lqx.spherical neuralqx.lqx.spherical.abstract_lqx_bosw_model neuralqx.lqx.spherical.core neuralqx.lqx.spherical.core.spherical_bosw neuralqx.lqx.spherical.spherical_bojowald_swiderski_interface neuralqx.lqx.wcl neuralqx.lqx.wcl.abstract_lqx_wcl_model neuralqx.lqx.wcl.core neuralqx.lqx.wcl.core.wcl_1d neuralqx.lqx.wcl.core.wcl_3d neuralqx.lqx.wcl.core.wcl_4d neuralqx.lqx.wcl.euclidean_interface neuralqx.nn neuralqx.nn.models neuralqx.nn.models.initialisers neuralqx.nn.projectors neuralqx.nn.projectors.group_projector neuralqx.nn.projectors.group_projector.group_projector neuralqx.nn.projectors.utils neuralqx.nn.projectors.vacuum neuralqx.nn.projectors.vacuum.novacuum_projector neuralqx.operators neuralqx.operators.coloring neuralqx.operators.computational neuralqx.operators.computational.Euclidean3d neuralqx.operators.computational.Euclidean3d.dispatch neuralqx.operators.computational.Euclidean3d.jax neuralqx.operators.computational.Euclidean3d.jax.vertex_TRC neuralqx.operators.computational.Euclidean3d.jax.volume neuralqx.operators.computational.Euclidean3d.numba neuralqx.operators.computational.Euclidean3d.numba.vertex_TRC neuralqx.operators.computational.Euclidean3d.numba.volume neuralqx.operators.computational.Euclidean4d neuralqx.operators.computational.Euclidean4d.dispatch neuralqx.operators.computational.Euclidean4d.jax neuralqx.operators.computational.Euclidean4d.jax.area neuralqx.operators.computational.Euclidean4d.jax.area_difference neuralqx.operators.computational.Euclidean4d.jax.gauss neuralqx.operators.computational.Euclidean4d.jax.vertex_TRC neuralqx.operators.computational.Euclidean4d.jax.volume neuralqx.operators.computational.Euclidean4d.numba neuralqx.operators.computational.Euclidean4d.numba.area neuralqx.operators.computational.Euclidean4d.numba.area_difference neuralqx.operators.computational.Euclidean4d.numba.gauss neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC neuralqx.operators.computational.Euclidean4d.numba.volume neuralqx.operators.computational.misc neuralqx.operators.computational.misc.dispatch neuralqx.operators.computational.misc.jax neuralqx.operators.computational.misc.numba neuralqx.operators.computational.qr neuralqx.operators.computational.qr.dispatch neuralqx.operators.computational.qr.jax neuralqx.operators.computational.qr.jax.annihilation neuralqx.operators.computational.qr.jax.creation neuralqx.operators.computational.qr.jax.euclidean_constraint neuralqx.operators.computational.qr.jax.flux neuralqx.operators.computational.qr.jax.lorentzian_constraint neuralqx.operators.computational.qr.numba neuralqx.operators.computational.qr.numba.annihilation neuralqx.operators.computational.qr.numba.creation neuralqx.operators.computational.qr.numba.euclidean_constraint neuralqx.operators.computational.qr.numba.flux neuralqx.operators.computational.qr.numba.lorentzian_constraint neuralqx.operators.computational.spherical neuralqx.operators.computational.spherical.dispatch neuralqx.operators.computational.spherical.jax neuralqx.operators.computational.spherical.numba neuralqx.operators.computational.wrappers neuralqx.operators.computational.wrappers.jax neuralqx.operators.computational.wrappers.numba neuralqx.operators.creation neuralqx.operators.holonomies neuralqx.operators.number neuralqx.operators.types neuralqx.operators.types.computational_operator neuralqx.operators.types.computational_operator.base neuralqx.operators.types.computational_operator.jax neuralqx.operators.types.functional_local_operator neuralqx.operators.types.functional_local_operator.base neuralqx.operators.types.functional_local_operator.compile_helpers neuralqx.operators.types.functional_local_operator.convert neuralqx.operators.types.functional_local_operator.helpers neuralqx.operators.types.functional_local_operator.jax neuralqx.operators.types.functional_local_operator.numba neuralqx.optimizer neuralqx.optimizer.builds neuralqx.optimizer.export neuralqx.optimizer.optimizer neuralqx.optimizer.schedule neuralqx.optimizer.solvers neuralqx.optimizer.solvers.builds neuralqx.optimizer.solvers.export neuralqx.optimizer.solvers.solvers neuralqx.optimizer.solvers.types neuralqx.optimizer.types neuralqx.profile neuralqx.profile.aggregate neuralqx.profile.decorators neuralqx.profile.profiler neuralqx.profile.summary neuralqx.samplers neuralqx.samplers.builds neuralqx.samplers.export neuralqx.samplers.rules neuralqx.samplers.sampler neuralqx.samplers.types neuralqx.solver neuralqx.solver.multi_state_solver neuralqx.solver.solver neuralqx.utils neuralqx.utils.deprecation neuralqx.utils.distributed neuralqx.utils.distributed.core neuralqx.utils.distributed.io neuralqx.utils.dtypes neuralqx.utils.dtypes.names neuralqx.utils.dtypes.policy neuralqx.utils.dtypes.runtime neuralqx.utils.errors neuralqx.utils.experimental neuralqx.utils.io neuralqx.utils.io.loggers neuralqx.utils.io.loggers.logger neuralqx.utils.io.printing neuralqx.utils.io.printing.nqxprinter neuralqx.utils.io.printing.types neuralqx.utils.io.runtime_loggers neuralqx.utils.io.runtime_loggers.deferred_runtime_log neuralqx.utils.jax neuralqx.utils.jax.djaxrun neuralqx.utils.jax.sharding neuralqx.utils.jit neuralqx.utils.misc neuralqx.utils.misc.arithmetic neuralqx.utils.misc.auth neuralqx.utils.misc.charges neuralqx.utils.misc.graph neuralqx.utils.misc.levi_civita neuralqx.utils.misc.modifiers neuralqx.utils.misc.triangulation neuralqx.utils.module neuralqx.utils.module.dependency_check neuralqx.utils.module.optional_dependency neuralqx.utils.module.version neuralqx.utils.module.version.coercion neuralqx.utils.module.version.module_version neuralqx.utils.module.version.neuralqx_version neuralqx.utils.module.version.version neuralqx.utils.module.version_check neuralqx.utils.numbers neuralqx.utils.parsing neuralqx.utils.serialization neuralqx.utils.symmetries neuralqx.utils.symmetries.graph_symmetries neuralqx.utils.symmetries.types neuralqx.utils.symmetries.types.automorphism neuralqx.utils.symmetries.types.cycle neuralqx.utils.symmetries.types.permutation neuralqx.utils.symmetries.types.symmetry neuralqx.utils.symmetries.types.utils neuralqx.utils.symmetries.utils neuralqx.vqs neuralqx.vqs.base neuralqx.vqs.mc neuralqx.vqs.mc.common neuralqx.vqs.mc.kernels neuralqx.vqs.mc.mc_state neuralqx.vqs.mc.mc_state.expect neuralqx.vqs.mc.mc_state.expect_chunked neuralqx.vqs.mc.mc_state.expect_forces neuralqx.vqs.mc.mc_state.expect_forces_chunked neuralqx.vqs.mc.mc_state.expect_grad neuralqx.vqs.mc.mc_state.expect_grad_chunked neuralqx.vqs.mc.mc_state.multi_state neuralqx.vqs.mc.mc_state.state neuralqx.vqs.mc.mc_state.utils

Momentum (in module neuralqx.optimizer.types) momentum (SGD attribute) Move (class in neuralqx.hilbert.u1.operations.moves) mpi_any() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) mpi_bcast() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) mpi_mean_jax() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) mpi_sum_jax() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) mu_dtype (AMSGrad attribute) MuLocal (class in neuralqx.samplers.types) MultiHamiltonianRule() (in module neuralqx.samplers.rules) MultiMCState (class in neuralqx.vqs) (class in neuralqx.vqs.mc) (class in neuralqx.vqs.mc.mc_state) (class in neuralqx.vqs.mc.mc_state.multi_state) MultiSolver (class in neuralqx.solver) (class in neuralqx.solver.multi_state_solver) MultiStateVMC (class in neuralqx.driver) (class in neuralqx.driver.multi_state_vmc) MuSampler (class in neuralqx.samplers) (class in neuralqx.samplers.rules)

N

n (Grid2D property) (HexagonalLatticeGraph property) N (HypercubeGraph property) n (TriangularLatticeGraph property) n_conn() (FunctionalLocalOperatorJax method) n_discard_per_chain (MCState property) (MultiMCState property) (STMultiMCState property) n_dual_edges (AbstractGraph property) n_dual_vertices (AbstractGraph property) n_edges (AbstractGraph property) (FreeEdgeFlipAllGauge attribute) (FreeEdgeFlipSingleGauge attribute) n_free (GaugeFixingTopo property) n_heads (STMultiSolver property), [1] n_minimal_loops (Graph property) n_operators (LocalOperatorBase property) n_parameters (MultiMCState property) (VariationalState property) n_point_function() (in module neuralqx.operators) (in module neuralqx.operators.coloring) n_samples (MCState property) (MultiMCState property) (STMultiMCState property) n_samples_per_rank (MCState property) (MultiMCState property) (STMultiMCState property) n_states (MultiMCState property) (STMultiMCState property) n_vertices (AbstractGraph property) name (AbstractGaugeGroup property) (Automorphism attribute) (Cholesky attribute) (CompiledOperator attribute), [1], [2], [3] (ConjugateGradient attribute) (Cycle attribute) (DefaultSolver attribute) (GroupProjectorDynamic attribute) (LU attribute) (NoVacuumProjector attribute) (Permutation attribute) (PseudoInverse attribute) (PseudoInverseSmooth attribute) (SingularValueDecomposition attribute) (SymbolicIRTerm attribute), [1], [2], [3] (SymbolicOperator attribute), [1], [2], [3] (U1GaugeGroup property) NDMinimalLoopHolonomy (class in neuralqx.operators.computational.misc.numba) NDMinimalLoopHolonomyJax (class in neuralqx.operators.computational.misc.jax) NDMinimalLoopHolonomyOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) NDU1Holonomy (class in neuralqx.operators.computational.misc.numba) NDU1HolonomyJax (class in neuralqx.operators.computational.misc.jax) NDU1HolonomyOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) nesterov (SGD attribute) neuralqx module neuralqx.callbacks module neuralqx.callbacks.converged_state_export module neuralqx.callbacks.live_monitoring module neuralqx.callbacks.pairwise_gradient_cosine_callback module neuralqx.configs module neuralqx.debug module neuralqx.driver module neuralqx.driver.abstract_variational_driver module neuralqx.driver.multi_state_vmc module neuralqx.driver.vmc module neuralqx.experimental module neuralqx.experimental.driver module, [1] neuralqx.experimental.driver.stmh_multi_vmc module, [1] neuralqx.experimental.hilbert module neuralqx.experimental.hilbert.su2 module neuralqx.experimental.hilbert.su2.index module neuralqx.experimental.hilbert.su2.layout module neuralqx.experimental.hilbert.su2.metadata module neuralqx.experimental.hilbert.su2.operations module neuralqx.experimental.nn module neuralqx.experimental.nn.projectors module neuralqx.experimental.nn.projectors.stmh module neuralqx.experimental.operators module, [1] neuralqx.experimental.operators.symbolic module, [1] neuralqx.experimental.operators.symbolic.compiler module neuralqx.experimental.operators.symbolic.compiler.cache module neuralqx.experimental.operators.symbolic.compiler.cache.store module neuralqx.experimental.operators.symbolic.compiler.compiler module neuralqx.experimental.operators.symbolic.compiler.core module neuralqx.experimental.operators.symbolic.compiler.core.artifact module neuralqx.experimental.operators.symbolic.compiler.core.context module neuralqx.experimental.operators.symbolic.compiler.core.options module neuralqx.experimental.operators.symbolic.compiler.core.pass_report module neuralqx.experimental.operators.symbolic.compiler.core.pipeline module neuralqx.experimental.operators.symbolic.compiler.core.signature module neuralqx.experimental.operators.symbolic.compiler.defaults module neuralqx.experimental.operators.symbolic.compiler.lowering module neuralqx.experimental.operators.symbolic.compiler.lowering.base module neuralqx.experimental.operators.symbolic.compiler.lowering.jax_lowerer module neuralqx.experimental.operators.symbolic.compiler.lowering.registry module neuralqx.experimental.operators.symbolic.compiler.passes module neuralqx.experimental.operators.symbolic.compiler.passes.analysis module neuralqx.experimental.operators.symbolic.compiler.passes.base module neuralqx.experimental.operators.symbolic.compiler.passes.fusion module neuralqx.experimental.operators.symbolic.compiler.passes.normalization module neuralqx.experimental.operators.symbolic.compiler.passes.validation module neuralqx.experimental.operators.symbolic.core module neuralqx.experimental.operators.symbolic.core.base module neuralqx.experimental.operators.symbolic.core.compiled module neuralqx.experimental.operators.symbolic.core.operator module neuralqx.experimental.operators.symbolic.core.sum module neuralqx.experimental.operators.symbolic.dsl module neuralqx.experimental.operators.symbolic.dsl.context module neuralqx.experimental.operators.symbolic.dsl.op module neuralqx.experimental.operators.symbolic.dsl.rewrite module neuralqx.experimental.operators.symbolic.dsl.selectors module neuralqx.experimental.operators.symbolic.ir module neuralqx.experimental.operators.symbolic.ir.expressions module neuralqx.experimental.operators.symbolic.ir.predicates module neuralqx.experimental.operators.symbolic.ir.program module neuralqx.experimental.operators.symbolic.ir.term module neuralqx.experimental.operators.symbolic.ir.update module neuralqx.experimental.operators.symbolic.ir.validate module neuralqx.experimental.solver module, [1] neuralqx.experimental.solver.stmh_solver module, [1] neuralqx.experimental.utils module neuralqx.experimental.utils.errors module neuralqx.experimental.utils.struct module neuralqx.experimental.utils.struct.base module neuralqx.experimental.utils.struct.fields module neuralqx.experimental.utils.struct.io module neuralqx.experimental.utils.struct.pytree module neuralqx.experimental.utils.struct.registry module neuralqx.experimental.vqs module, [1] neuralqx.experimental.vqs.mc module, [1] neuralqx.experimental.vqs.mc.mc_state module neuralqx.experimental.vqs.mc.mc_state.expect_grad_biadjoint module neuralqx.experimental.vqs.mc.mc_state.stmh_state module neuralqx.experimental.vqs.mc.mc_state.utils module neuralqx.gauge_groups module neuralqx.gauge_groups.abstract_gauge_group_interface module neuralqx.gauge_groups.u1 module neuralqx.gauge_groups.u1.u1_gauge_group module neuralqx.graph module neuralqx.graph.circular_ladder module neuralqx.graph.core module neuralqx.graph.core.abstract_graph module neuralqx.graph.core.graph_handler module neuralqx.graph.core.utils module neuralqx.graph.core.utils.build module neuralqx.graph.core.utils.checks module neuralqx.graph.core.utils.embedding module neuralqx.graph.core.utils.plotting module neuralqx.graph.core.utils.sign_utils module neuralqx.graph.core.utils.signs module neuralqx.graph.dodecahedral module neuralqx.graph.graph module neuralqx.graph.grid_2d module neuralqx.graph.half_ladder_graph module neuralqx.graph.hexagonal_lattice module neuralqx.graph.hypercube module neuralqx.graph.k5_graph module neuralqx.graph.ladder_graph module neuralqx.graph.single_vertex module neuralqx.graph.triangular_lattice module neuralqx.hilbert module neuralqx.hilbert.abstract_hilbert_core module neuralqx.hilbert.abstract_hilbert_interface module neuralqx.hilbert.constraints module neuralqx.hilbert.constraints.u1_constraint module neuralqx.hilbert.constraints.utils module neuralqx.hilbert.u1 module neuralqx.hilbert.u1.constrained_core module neuralqx.hilbert.u1.index module neuralqx.hilbert.u1.index.enumerator module neuralqx.hilbert.u1.index.mapping module neuralqx.hilbert.u1.index.utils module neuralqx.hilbert.u1.interface module neuralqx.hilbert.u1.layout module neuralqx.hilbert.u1.layout.strided module neuralqx.hilbert.u1.operations module neuralqx.hilbert.u1.operations.flip module neuralqx.hilbert.u1.operations.moves module neuralqx.hilbert.u1.operations.random module neuralqx.hilbert.u1.unconstrained_core module neuralqx.hilbert.utils module neuralqx.hilbert.utils.index module neuralqx.hilbert.utils.index.dispatch module neuralqx.hilbert.utils.layout module neuralqx.lqx module neuralqx.lqx.abstract_lqx_interface module neuralqx.lqx.abstract_lqx_model module neuralqx.lqx.qr module neuralqx.lqx.qr.abstract_single_vertex_qr_model module neuralqx.lqx.qr.core module neuralqx.lqx.qr.core.single_vertex_qrlg module neuralqx.lqx.qr.single_vertex_qr_interface module neuralqx.lqx.spherical module neuralqx.lqx.spherical.abstract_lqx_bosw_model module neuralqx.lqx.spherical.core module neuralqx.lqx.spherical.core.spherical_bosw module neuralqx.lqx.spherical.spherical_bojowald_swiderski_interface module neuralqx.lqx.wcl module neuralqx.lqx.wcl.abstract_lqx_wcl_model module neuralqx.lqx.wcl.core module neuralqx.lqx.wcl.core.wcl_1d module neuralqx.lqx.wcl.core.wcl_3d module neuralqx.lqx.wcl.core.wcl_4d module neuralqx.lqx.wcl.euclidean_interface module neuralqx.nn module neuralqx.nn.models module neuralqx.nn.models.initialisers module neuralqx.nn.projectors module

neuralqx.nn.projectors.group_projector module neuralqx.nn.projectors.group_projector.group_projector module neuralqx.nn.projectors.utils module neuralqx.nn.projectors.vacuum module neuralqx.nn.projectors.vacuum.novacuum_projector module neuralqx.operators module neuralqx.operators.coloring module neuralqx.operators.computational module neuralqx.operators.computational.Euclidean3d module neuralqx.operators.computational.Euclidean3d.dispatch module neuralqx.operators.computational.Euclidean3d.jax module neuralqx.operators.computational.Euclidean3d.jax.vertex_TRC module neuralqx.operators.computational.Euclidean3d.jax.volume module neuralqx.operators.computational.Euclidean3d.numba module neuralqx.operators.computational.Euclidean3d.numba.vertex_TRC module neuralqx.operators.computational.Euclidean3d.numba.volume module neuralqx.operators.computational.Euclidean4d module neuralqx.operators.computational.Euclidean4d.dispatch module neuralqx.operators.computational.Euclidean4d.jax module neuralqx.operators.computational.Euclidean4d.jax.area module neuralqx.operators.computational.Euclidean4d.jax.area_difference module neuralqx.operators.computational.Euclidean4d.jax.gauss module neuralqx.operators.computational.Euclidean4d.jax.vertex_TRC module neuralqx.operators.computational.Euclidean4d.jax.volume module neuralqx.operators.computational.Euclidean4d.numba module neuralqx.operators.computational.Euclidean4d.numba.area module neuralqx.operators.computational.Euclidean4d.numba.area_difference module neuralqx.operators.computational.Euclidean4d.numba.gauss module neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC module neuralqx.operators.computational.Euclidean4d.numba.volume module neuralqx.operators.computational.misc module neuralqx.operators.computational.misc.dispatch module neuralqx.operators.computational.misc.jax module neuralqx.operators.computational.misc.numba module neuralqx.operators.computational.qr module neuralqx.operators.computational.qr.dispatch module neuralqx.operators.computational.qr.jax module neuralqx.operators.computational.qr.jax.annihilation module neuralqx.operators.computational.qr.jax.creation module neuralqx.operators.computational.qr.jax.euclidean_constraint module neuralqx.operators.computational.qr.jax.flux module neuralqx.operators.computational.qr.jax.lorentzian_constraint module neuralqx.operators.computational.qr.numba module neuralqx.operators.computational.qr.numba.annihilation module neuralqx.operators.computational.qr.numba.creation module neuralqx.operators.computational.qr.numba.euclidean_constraint module neuralqx.operators.computational.qr.numba.flux module neuralqx.operators.computational.qr.numba.lorentzian_constraint module neuralqx.operators.computational.spherical module neuralqx.operators.computational.spherical.dispatch module neuralqx.operators.computational.spherical.jax module neuralqx.operators.computational.spherical.numba module neuralqx.operators.computational.wrappers module neuralqx.operators.computational.wrappers.jax module neuralqx.operators.computational.wrappers.numba module neuralqx.operators.creation module neuralqx.operators.holonomies module neuralqx.operators.number module neuralqx.operators.types module neuralqx.operators.types.computational_operator module neuralqx.operators.types.computational_operator.base module neuralqx.operators.types.computational_operator.jax module neuralqx.operators.types.functional_local_operator module neuralqx.operators.types.functional_local_operator.base module neuralqx.operators.types.functional_local_operator.compile_helpers module neuralqx.operators.types.functional_local_operator.convert module neuralqx.operators.types.functional_local_operator.helpers module neuralqx.operators.types.functional_local_operator.jax module neuralqx.operators.types.functional_local_operator.numba module neuralqx.optimizer module neuralqx.optimizer.builds module neuralqx.optimizer.export module neuralqx.optimizer.optimizer module neuralqx.optimizer.schedule module neuralqx.optimizer.solvers module neuralqx.optimizer.solvers.builds module neuralqx.optimizer.solvers.export module neuralqx.optimizer.solvers.solvers module neuralqx.optimizer.solvers.types module neuralqx.optimizer.types module neuralqx.profile module neuralqx.profile.aggregate module neuralqx.profile.decorators module neuralqx.profile.profiler module neuralqx.profile.summary module neuralqx.samplers module neuralqx.samplers.builds module neuralqx.samplers.export module neuralqx.samplers.rules module neuralqx.samplers.sampler module neuralqx.samplers.types module neuralqx.solver module neuralqx.solver.multi_state_solver module neuralqx.solver.solver module neuralqx.utils module neuralqx.utils.deprecation module neuralqx.utils.distributed module neuralqx.utils.distributed.core module neuralqx.utils.distributed.io module neuralqx.utils.dtypes module neuralqx.utils.dtypes.names module neuralqx.utils.dtypes.policy module neuralqx.utils.dtypes.runtime module neuralqx.utils.errors module neuralqx.utils.experimental module neuralqx.utils.io module neuralqx.utils.io.loggers module neuralqx.utils.io.loggers.logger module neuralqx.utils.io.printing module neuralqx.utils.io.printing.nqxprinter module neuralqx.utils.io.printing.types module neuralqx.utils.io.runtime_loggers module neuralqx.utils.io.runtime_loggers.deferred_runtime_log module neuralqx.utils.jax module neuralqx.utils.jax.djaxrun module neuralqx.utils.jax.sharding module neuralqx.utils.jit module neuralqx.utils.misc module neuralqx.utils.misc.arithmetic module neuralqx.utils.misc.auth module neuralqx.utils.misc.charges module neuralqx.utils.misc.graph module neuralqx.utils.misc.levi_civita module neuralqx.utils.misc.modifiers module neuralqx.utils.misc.triangulation module neuralqx.utils.module module neuralqx.utils.module.dependency_check module neuralqx.utils.module.optional_dependency module neuralqx.utils.module.version module neuralqx.utils.module.version.coercion module neuralqx.utils.module.version.module_version module neuralqx.utils.module.version.neuralqx_version module neuralqx.utils.module.version.version module neuralqx.utils.module.version_check module neuralqx.utils.numbers module neuralqx.utils.parsing module neuralqx.utils.serialization module neuralqx.utils.symmetries module neuralqx.utils.symmetries.graph_symmetries module neuralqx.utils.symmetries.types module neuralqx.utils.symmetries.types.automorphism module neuralqx.utils.symmetries.types.cycle module neuralqx.utils.symmetries.types.permutation module neuralqx.utils.symmetries.types.symmetry module neuralqx.utils.symmetries.types.utils module neuralqx.utils.symmetries.utils module neuralqx.vqs module neuralqx.vqs.base module neuralqx.vqs.mc module neuralqx.vqs.mc.common module neuralqx.vqs.mc.kernels module neuralqx.vqs.mc.mc_state module neuralqx.vqs.mc.mc_state.expect module neuralqx.vqs.mc.mc_state.expect_chunked module neuralqx.vqs.mc.mc_state.expect_forces module neuralqx.vqs.mc.mc_state.expect_forces_chunked module neuralqx.vqs.mc.mc_state.expect_grad module neuralqx.vqs.mc.mc_state.expect_grad_chunked module neuralqx.vqs.mc.mc_state.multi_state module neuralqx.vqs.mc.mc_state.state module neuralqx.vqs.mc.mc_state.utils module neuralqxError, [1] NeuralqxVersion (class in neuralqx.utils.module.version) (class in neuralqx.utils.module.version.neuralqx_version) (class in neuralqx.utils.module.version_check) neuralqxWarning nk_graph (AbstractGraph property) (Graph property) node_fields() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) (Struct class method), [1] NonExistentNonPlanarEdgesError NonExistentNonPlanarVerticesError NonHermitianInverseCostError nonplanar_edges (AbstractGraph property) nonplanar_vertex_mapping (AbstractGraph property) nonplanar_vertex_to_index() (AbstractGraph method) NonplanarSigns (class in neuralqx.graph.core.utils.sign_utils) normalize_pspec() (in module neuralqx.utils.jax.sharding) normalize_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.version) (in module neuralqx.utils.module.version_check) NoVacuumProjector (class in neuralqx.nn.projectors.vacuum) (class in neuralqx.nn.projectors.vacuum.novacuum_projector) NQXPrinter (class in neuralqx.utils.io.printing) (class in neuralqx.utils.io.printing.nqxprinter) Number (class in neuralqx.operators.computational.misc.numba) number() (AbstractLqxSphericalBoSw method) (LqxBojowaldSwiderski method) (LqxSphericalBoSw method) (LqxWCL1D method) (LqxWclModel method) NumberJax (class in neuralqx.operators.computational.misc.jax) NumberOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) numbers_to_states() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (HilbertStateEnumerator method) (in module neuralqx.hilbert.u1.index) (in module neuralqx.hilbert.u1.index.mapping) (in module neuralqx.hilbert.utils.index) (in module neuralqx.hilbert.utils.index.dispatch) nx_graph (AbstractGraph property) (Graph property)

O

old_name_deprecation() (in module neuralqx.utils.deprecation) OneMinusHalfHolonomy (class in neuralqx.operators.computational.misc.numba) OneMinusHalfHolonomyJax (class in neuralqx.operators.computational.misc.jax) OneMinusHalfHolonomyOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) ones_complex() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) ones_real() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) op (AmplitudeExpr attribute), [1], [2], [3] (PredicateExpr attribute), [1], [2], [3] opaque_fields() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) (Struct class method), [1]

operator_ir_fingerprint (SymbolicCompilationSignature attribute), [1], [2], [3], [4] operator_name (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] (SymbolicOperatorIR attribute), [1], [2], [3] operators (LocalOperatorBase property) ops (UpdateProgram attribute), [1], [2], [3] OptaxSchedule (class in neuralqx.optimizer.types) Optimizer (class in neuralqx.optimizer) (class in neuralqx.optimizer.optimizer) optimizer (Optimizer property) options_signature (SymbolicCompilationSignature attribute), [1], [2], [3], [4] orientation (SingleVertexGraph property) OrientationValenceMismatchError OutOfRangeIndexError overlap_matrix() (MultiMCState method)

P

p_ops (MetropolisMultiHamiltonian attribute) (PTHamiltonianList attribute) pack_internals() (in module neuralqx.operators.types.functional_local_operator.compile_helpers) pack_internals_jax() (in module neuralqx.operators.types.functional_local_operator.compile_helpers) parameter_name_deprecation() (in module neuralqx.utils.deprecation) parameters (MultiMCState property) (STMultiMCState property) (VariationalState property) params (UpdateOp attribute), [1], [2], [3] params_are_complex() (in module neuralqx.experimental.vqs.mc.mc_state.utils) parent (GroupProjectorDynamic attribute) (NoVacuumProjector attribute) parse_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.version) (in module neuralqx.utils.module.version_check) partial_deserialize() (in module neuralqx.utils.serialization) partial_serialize() (in module neuralqx.utils.serialization) pass_name (SymbolicPassReport attribute), [1] pass_reports (SymbolicCompiledArtifact attribute), [1], [2], [3], [4] patch (NeuralqxVersion property) (Version property) patch_attr() (in module neuralqx.profile) (in module neuralqx.profile.decorators) patch_method() (in module neuralqx.profile) (in module neuralqx.profile.decorators) penalty_expectation_gradient() (in module neuralqx.operators) penalty_expectation_value() (in module neuralqx.operators) penalty_is_linear() (in module neuralqx.operators) penalty_linear_scale() (in module neuralqx.operators) penalty_local_value_coefficients() (in module neuralqx.operators) PenaltyCost (class in neuralqx.operators) periodic (Grid2D property) (HexagonalLatticeGraph property) (TriangularLatticeGraph property) perms_edge (GroupProjectorDynamic attribute) perms_full (GroupProjectorDynamic attribute) Permutation (class in neuralqx.utils.symmetries.types) (class in neuralqx.utils.symmetries.types.permutation) permutations (GraphSymmetries property) permute() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] plaquette_flip() (ConstrainedHilbertU1Core method) PlaquetteFlipAllGauge (class in neuralqx.hilbert.u1.operations.moves) PlaquetteFlipSingleGauge (class in neuralqx.hilbert.u1.operations.moves) plot_observables() (Solver method) plot_results() (MultiSolver method) (Solver method) (STMultiSolver method), [1] plus_key() (in module neuralqx.utils.misc.arithmetic)

preconditioner (VMC property) predicate (SymbolicIRTerm attribute), [1], [2], [3] PredicateExpr (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.predicates) preprocess_constraint() (in module neuralqx.hilbert.constraints.utils) preprocess_constraints_for_jax() (in module neuralqx.hilbert.constraints.utils) pretty_format_constraints() (in module neuralqx.hilbert.constraints.utils) print() (NQXPrinter method) print0() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) print_overlap_matrix() (MultiMCState method) print_signs() (GraphHandler method) PrintEntry (class in neuralqx.utils.io.printing.types) PrintHandler (class in neuralqx.utils.io.printing.types) process_count() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) process_index() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) Product (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.numba) ProductJax (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.jax) profile() (in module neuralqx.profile) (in module neuralqx.profile.decorators) profile_call() (in module neuralqx.profile) (in module neuralqx.profile.decorators) Profiler (class in neuralqx.profile) (class in neuralqx.profile.profiler) project() (MCState method) (MultiMCState method) projector() (in module neuralqx.nn.projectors.group_projector) (in module neuralqx.nn.projectors.group_projector.group_projector) (in module neuralqx.nn.projectors.vacuum) (in module neuralqx.nn.projectors.vacuum.novacuum_projector) promote_constant_for_op_dtype() (in module neuralqx.utils.numbers) propose() (ConstrainedHilbertU1Core method) PseudoInverse (class in neuralqx.optimizer.solvers.types) PseudoInverseSmooth (class in neuralqx.optimizer.solvers.types) PTCommon (class in neuralqx.samplers.types) PTExchange (class in neuralqx.samplers.types) PTHamiltonianList (class in neuralqx.samplers.types) PTLocal (class in neuralqx.samplers.types) PTRandomU1Gauge (class in neuralqx.samplers.types) PTU1Gauge (class in neuralqx.samplers.types) PTU1Plaquette (class in neuralqx.samplers.types) PTWeighted (class in neuralqx.samplers.types) python_call_trace() (in module neuralqx.profile) (in module neuralqx.profile.decorators) PyTreeTypeSpec (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.pytree)

Q

q_max (AbstractHilbertSpace property) q_min (AbstractHilbertSpace property) q_step (AbstractHilbertSpace property) QRAnnihilation (class in neuralqx.operators.computational.qr.numba) (class in neuralqx.operators.computational.qr.numba.annihilation) QRAnnihilationJax (class in neuralqx.operators.computational.qr.jax) (class in neuralqx.operators.computational.qr.jax.annihilation) QRAnnihilationOperator() (in module neuralqx.operators.computational.qr) (in module neuralqx.operators.computational.qr.dispatch) QRCreation (class in neuralqx.operators.computational.qr.numba) (class in neuralqx.operators.computational.qr.numba.creation) QRCreationJax (class in neuralqx.operators.computational.qr.jax) (class in neuralqx.operators.computational.qr.jax.creation)

QRCreationOperator() (in module neuralqx.operators.computational.qr) (in module neuralqx.operators.computational.qr.dispatch) QRFlux (class in neuralqx.operators.computational.qr.numba) (class in neuralqx.operators.computational.qr.numba.flux) QRFluxJax (class in neuralqx.operators.computational.qr.jax) (class in neuralqx.operators.computational.qr.jax.flux) QRFluxOperator() (in module neuralqx.operators.computational.qr) (in module neuralqx.operators.computational.qr.dispatch) quantum_geometric_tensor() (MCState method) (MultiMCState method) (VariationalState method)

R

random_state() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (UnconstrainedHilbertU1Core method) RandomEmbeddingForPlanarGraphWarning randomize_graph_coordinates() (in module neuralqx.graph.core.utils.embedding) RandomU1Gauge (class in neuralqx.samplers.types) RandomU1GaugeSampler (class in neuralqx.samplers) (class in neuralqx.samplers.rules) rank (RuntimeInfo attribute), [1] RankStat (class in neuralqx.profile.aggregate) rationale (DependencyPolicy attribute) real (DTypePolicy attribute), [1] reduce_permutations() (in module neuralqx.utils.misc.graph) refresh_if_needed() (Profiler method) refresh_settings() (in module neuralqx.debug) register_attrs_type() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.pytree) register_class() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) register_class_ref() (in module neuralqx.experimental.utils.struct.registry) register_pytree_type() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.pytree) reimpose_gauge_fixing() (ConstrainedHilbertU1Core method) reject_outliers() (in module neuralqx.solver.solver) relabel_edges_to_nonplanar() (in module neuralqx.graph.core.utils.build) relabel_nx_edges() (in module neuralqx.graph.core.utils.build) release (NeuralqxVersion property) (Version property)

render_panels() (in module neuralqx.utils.jax.djaxrun) render_summary_text() (in module neuralqx.profile.summary) reorder_edge_triplet_key() (in module neuralqx.utils.misc.graph) reorder_edges() (in module neuralqx.graph.core.utils.plotting) reorder_minimal_loop_key() (in module neuralqx.utils.misc.graph) replicate_sharding() (in module neuralqx.utils.jax) (in module neuralqx.utils.jax.sharding) replicated_sharding() (in module neuralqx.utils.jax.sharding) required_kwargs() (in module neuralqx.utils.parsing) reset() (MCState method) (MultiMCState method) (STMultiMCState method) (VariationalState method) ResNet (class in neuralqx.nn.models) resolve_class() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.registry) resolve_factored_local_kernel() (in module neuralqx.vqs.mc.kernels) resolve_pytree_spec() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.pytree) RevNet (class in neuralqx.nn.models) RevNetAuto (class in neuralqx.nn.models) RevNetComplex (class in neuralqx.nn.models) RMSProp (class in neuralqx.optimizer.types) run() (AbstractVariationalDriver method) (Solver method) runtime_info() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.core) RuntimeInfo (class in neuralqx.utils.distributed) (class in neuralqx.utils.distributed.core)

S

s() (AbstractLqxSVQRLGModel method) (LqxSingleVertexQR method) (LqxSVQRLG method) safe_replicate_for_io() (in module neuralqx.utils.distributed) (in module neuralqx.utils.distributed.io) same_treedef() (in module neuralqx.experimental.vqs.mc.mc_state.utils) sample() (MCState method) (MultiMCState method) (STMultiMCState method) Sampler (class in neuralqx.samplers) (class in neuralqx.samplers.sampler) sampler (MCState property) (MultiMCState property) (Sampler property) (STMultiMCState property) sampler_g (Sampler property) sampler_state (MultiMCState property) samples (MCState property) (MultiMCState property) (STMultiMCState property) sanitize() (Logger method) save_multi_partial() (in module neuralqx.utils.serialization) save_to_file() (in module neuralqx.utils.serialization) (SerializerEngine static method) Scaled (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.numba) ScaledJax (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.jax) scatter() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] scope (GroupProjectorDynamic attribute) (NoVacuumProjector attribute) section() (in module neuralqx.profile) (in module neuralqx.profile.decorators) (Profiler method), [1] SerializationError serialize() (DeferredRuntimeLog method) (in module neuralqx.utils.serialization) (SerializerEngine static method) serialize_MCState() (in module neuralqx.vqs.mc.mc_state.state) serialize_MultiMCState() (in module neuralqx.solver) (in module neuralqx.solver.multi_state_solver) serialize_STMultiMCState() (in module neuralqx.experimental.solver.stmh_solver), [1] SerializerEngine (class in neuralqx.utils.serialization) set_deferred_accumulation() (DeferredRuntimeLog method) set_network() (MultiSolver method) (Solver method) (STMultiSolver method), [1] set_optimizer() (Solver method) set_sampler() (Solver method) SGD (class in neuralqx.optimizer.types) sharded_sharding_1d() (in module neuralqx.utils.jax.sharding) shift() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] shift_loop() (in module neuralqx.lqx.wcl.core.wcl_3d) shift_mod() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] shifted_get_quantum_number() (in module neuralqx.operators) (in module neuralqx.operators.number) shutdown() (LiveMonitor method) (LiveMonitoringCallback method) sign_data() (Authenticator static method) signs (AbstractGraph property) SingleTrunkMultiHeadLogPsi (class in neuralqx.experimental.nn.projectors.stmh) SingleTrunkMultiHeadVMC (class in neuralqx.experimental.driver), [1] (class in neuralqx.experimental.driver.stmh_multi_vmc), [1] SingleVertexGraph (class in neuralqx.graph) (class in neuralqx.graph.single_vertex) SingularValueDecomposition (class in neuralqx.optimizer.solvers.types) site() (ExpressionContext method), [1], [2] (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.selectors) site_to_edge() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (UnconstrainedHilbertU1Core method) SiteSelector (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.dsl) (class in neuralqx.experimental.operators.symbolic.dsl.selectors) size (AbstractHilbertInterface property) (AbstractHilbertSpace property) (LocalOperatorBase property) Solver (class in neuralqx.solver) (class in neuralqx.solver.solver) solver (Solvers property) SolverConfig (class in neuralqx.optimizer.solvers.types) Solvers (class in neuralqx.optimizer.solvers) (class in neuralqx.optimizer.solvers.solvers) source (Version property) spacetime_dimensions (AbstractLqxInterface property) (AbstractLqxModel property) spherical_Ex() (AbstractLqxSphericalBoSw method) (LqxSphericalBoSw method) spherical_hamiltonian_on_vertex() (LqxSphericalBoSw method) spherical_volume() (AbstractLqxSphericalBoSw method) (LqxSphericalBoSw method) SphericalEx (class in neuralqx.operators.computational.spherical.numba) SphericalExJax (class in neuralqx.operators.computational.spherical.jax) SphericalExOperator() (in module neuralqx.operators.computational.spherical) (in module neuralqx.operators.computational.spherical.dispatch) SphericalVertexConstraintBojowaldSwiderski (class in neuralqx.operators.computational.spherical.numba) SphericalVertexConstraintBojowaldSwiderskiAdjoint (class in neuralqx.operators.computational.spherical.numba) SphericalVertexConstraintBojowaldSwiderskiAdjointFast (class in neuralqx.operators.computational.spherical.numba) SphericalVertexConstraintBojowaldSwiderskiFast (class in neuralqx.operators.computational.spherical.numba) SphericalVertexConstraintBojowaldSwiderskiJax (class in neuralqx.operators.computational.spherical.jax) SphericalVertexConstraintBojowaldSwiderskiJaxAdjoint (class in neuralqx.operators.computational.spherical.jax) SphericalVertexConstraintBojowaldSwiderskiOperator() (in module neuralqx.operators.computational.spherical) (in module neuralqx.operators.computational.spherical.dispatch) SphericalVolume (class in neuralqx.operators.computational.spherical.numba) SphericalVolumeJax (class in neuralqx.operators.computational.spherical.jax) SphericalVolumeOperator() (in module neuralqx.operators.computational.spherical) (in module neuralqx.operators.computational.spherical.dispatch) sqrt_volume() (LqxWCL3D method) (LqxWCL4D method)

SqrtVolumeOperator() (in module neuralqx.operators.computational.Euclidean3d) (in module neuralqx.operators.computational.Euclidean3d.dispatch) squared_volume() (LqxWCL3D method) (LqxWCL4D method) SquaredVolumeOperator() (in module neuralqx.operators.computational.Euclidean3d) (in module neuralqx.operators.computational.Euclidean3d.dispatch) staircase (ExponentialDecay attribute) StateExportCallback (class in neuralqx.callbacks) (class in neuralqx.callbacks.converged_state_export) states (MultiStateVMC property) (SingleTrunkMultiHeadVMC property), [1] states_to_numbers() (AbstractHilbertInterface method) (AbstractHilbertSpace method) (HilbertStateEnumerator method) (in module neuralqx.hilbert.u1.index) (in module neuralqx.hilbert.u1.index.mapping) (in module neuralqx.hilbert.utils.index) (in module neuralqx.hilbert.utils.index.dispatch) static_fields() (in module neuralqx.experimental.utils.struct) (in module neuralqx.experimental.utils.struct.base) (Struct class method), [1] step (AbstractHilbertSpace property) step() (in module neuralqx.profile) (in module neuralqx.profile.decorators) (Profiler method), [1] STMHAllHeadsView (class in neuralqx.experimental.nn.projectors.stmh) STMHHeadView (class in neuralqx.experimental.nn.projectors.stmh) STMultiMCState (class in neuralqx.experimental.vqs), [1] (class in neuralqx.experimental.vqs.mc.mc_state) (class in neuralqx.experimental.vqs.mc.mc_state.stmh_state) STMultiSolver (class in neuralqx.experimental.solver), [1] (class in neuralqx.experimental.solver.stmh_solver), [1] str_edge_parser() (in module neuralqx.utils.parsing) strict_type() (in module neuralqx.utils.parsing) strict_validation (SymbolicCompilerOptions attribute), [1], [2], [3], [4] StridedGaugeCopyLayout (class in neuralqx.hilbert.u1.layout) (class in neuralqx.hilbert.u1.layout.strided) Struct (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.base) StructABCMeta (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.base) StructError StructMeta (class in neuralqx.experimental.utils.struct) (class in neuralqx.experimental.utils.struct.base) SU2ChannelConstructionFailureError SU2ExactEnumerationTooLargeError SU2GraphMetadata (class in neuralqx.experimental.hilbert.su2.metadata) SU2UnconstrainedNotImplementedError SuboptimalOperatorForGPUWarning Sum (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.numba) SumJax (class in neuralqx.operators.computational.wrappers) (class in neuralqx.operators.computational.wrappers.jax) summary_dict() (Profiler method) swap() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] symbol() (AmplitudeExpr class method), [1] (ExpressionContext method), [1], [2] (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.selectors) SymbolicCacheKey (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.signature) SymbolicCompilationContext (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.context) SymbolicCompilationSignature (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.signature) SymbolicCompiledArtifact (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.artifact) SymbolicCompiler (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.compiler) SymbolicCompilerError SymbolicCompilerOptions (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.compiler) (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.options) SymbolicFanoutAnalysisPass (class in neuralqx.experimental.operators.symbolic.compiler.passes) (class in neuralqx.experimental.operators.symbolic.compiler.passes.analysis) SymbolicFusionPass (class in neuralqx.experimental.operators.symbolic.compiler.passes) (class in neuralqx.experimental.operators.symbolic.compiler.passes.fusion) SymbolicIRTerm (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.term) SymbolicLowererRegistry (class in neuralqx.experimental.operators.symbolic.compiler.lowering) (class in neuralqx.experimental.operators.symbolic.compiler.lowering.registry) SymbolicNormalizationPass (class in neuralqx.experimental.operators.symbolic.compiler.passes) (class in neuralqx.experimental.operators.symbolic.compiler.passes.normalization) SymbolicOperator (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.core) (class in neuralqx.experimental.operators.symbolic.core.operator) SymbolicOperatorDefinitionError SymbolicOperatorExecutionError SymbolicOperatorIR (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.program) SymbolicOperatorIRValidationError SymbolicOperatorSum (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.core) (class in neuralqx.experimental.operators.symbolic.core.sum) SymbolicPassPipeline (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.pipeline) SymbolicPassReport (class in neuralqx.experimental.operators.symbolic.compiler.core) (class in neuralqx.experimental.operators.symbolic.compiler.core.pass_report) SymbolicValidationPass (class in neuralqx.experimental.operators.symbolic.compiler.passes) (class in neuralqx.experimental.operators.symbolic.compiler.passes.validation) Symmetry (class in neuralqx.utils.symmetries.types.symmetry) symmetry_to_index_perm() (in module neuralqx.nn.projectors.utils) sync_enabled() (Profiler method) SystemInfoManager (class in neuralqx.utils)

T

tagged() (in module neuralqx.debug) taylor_expanded_volume() (LqxWCL3D method) (LqxWCL4D method) terms (SymbolicOperatorIR attribute), [1], [2], [3] thiemann_quadratic_constraint() (LqxWCL method) (LqxWCL1D method) (LqxWCL3D method) (LqxWCL4D method) (LqxWclModel method) thiemann_regularized_constraint() (LqxWCL3D method) (LqxWCL4D method) ThiemannRegularisedVertexConstraint() (in module neuralqx.operators.computational.Euclidean4d.numba) (in module neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) ThiemannRegularisedVertexConstraint3d() (in module neuralqx.operators.computational.Euclidean3d.numba) (in module neuralqx.operators.computational.Euclidean3d.numba.vertex_TRC) ThiemannRegularisedVertexConstraint3dJax() (in module neuralqx.operators.computational.Euclidean3d.jax) (in module neuralqx.operators.computational.Euclidean3d.jax.vertex_TRC) ThiemannRegularisedVertexConstraintFast() (in module neuralqx.operators.computational.Euclidean4d.numba) (in module neuralqx.operators.computational.Euclidean4d.numba.vertex_TRC) ThiemannRegularisedVertexConstraintJax() (in module neuralqx.operators.computational.Euclidean4d.jax) (in module neuralqx.operators.computational.Euclidean4d.jax.vertex_TRC) ThiemannRegularisedVertexConstraintOperator() (in module neuralqx.operators.computational.Euclidean3d) (in module neuralqx.operators.computational.Euclidean3d.dispatch) (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) timeit() (in module neuralqx.debug) tiny_hilbert (AbstractHilbertInterface property) (AbstractHilbertSpace property) tiny_size (AbstractHilbertInterface property) (AbstractHilbertSpace property) to_array() (MCState method) (MultiMCState method) (STMultiMCState method) (VariationalState method) to_array_numpy() (in module neuralqx.vqs.mc.mc_state.utils) (MCState method) to_dense() (ComputationalOperator method) to_group_averaged() (MCState method) (MultiMCState method) to_jax_operator() (FunctionalLocalOperator method) to_numba_operator() (FunctionalLocalOperatorJax method) to_pauli_strings() (FunctionalLocalOperatorJax method) (LocalOperatorBase method) to_python() (in module neuralqx.experimental.utils.struct.io)

to_python_scalar() (in module neuralqx.utils.numbers) to_qobj() (VariationalState method) to_sparse() (ComputationalOperator method) to_state_dict() (in module neuralqx.experimental.utils.struct.io) trace() (in module neuralqx.debug) trace_events() (Profiler method) transition_begin (LinearDecay attribute) transpose() (LocalOperatorBase method) tree_add() (in module neuralqx.experimental.vqs.mc.mc_state.utils) tree_add_scaled() (in module neuralqx.experimental.vqs.mc.mc_state.utils) tree_flatten() (AreaDifferenceSquaredOperatorJax method) (AreaDifferenceSquaredSurfacesOperatorJax method) (AreaOperatorJax method) (Euclidean3dVertexTRCOperatorJax method) (EuclideanConstraintJax method) (EuclideanVertexTRCOperatorJax method) (FunctionalLocalOperatorJax method) (GaussConstraintOperatorJax method) (LorentzianConstraintJax method) (QRAnnihilationJax method) (QRCreationJax method) (QRFluxJax method) (VolumeOperatorJax method), [1] tree_scale() (in module neuralqx.experimental.vqs.mc.mc_state.utils) tree_unflatten() (AreaDifferenceSquaredOperatorJax class method) (AreaDifferenceSquaredSurfacesOperatorJax class method) (AreaOperatorJax class method) (Euclidean3dVertexTRCOperatorJax class method) (EuclideanConstraintJax class method) (EuclideanVertexTRCOperatorJax class method) (FunctionalLocalOperatorJax class method) (GaussConstraintOperatorJax class method) (LorentzianConstraintJax class method) (QRAnnihilationJax class method) (QRCreationJax class method) (QRFluxJax class method) (VolumeOperatorJax class method), [1] tree_zeros_like() (in module neuralqx.experimental.vqs.mc.mc_state.utils) TriangularLatticeGraph (class in neuralqx.graph) (class in neuralqx.graph.triangular_lattice) Triangulation (class in neuralqx.utils.misc.triangulation) try_parse() (Version class method) try_parse_version() (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version.version) (in module neuralqx.utils.module.version_check)

U

U1ConstrainedStateEnumerator (class in neuralqx.hilbert.u1.index) (class in neuralqx.hilbert.u1.index.enumerator) U1Constraint (class in neuralqx.hilbert.constraints) (class in neuralqx.hilbert.constraints.u1_constraint) U1Gauge (class in neuralqx.samplers.types) U1GaugeGroup (class in neuralqx.gauge_groups.u1), [1] (class in neuralqx.gauge_groups.u1.u1_gauge_group) U1GaugeNonzero (class in neuralqx.samplers.types) U1GaugeSampler (class in neuralqx.samplers) (class in neuralqx.samplers.rules) U1GaugeSamplerNonzero (class in neuralqx.samplers.rules) U1Holonomy (class in neuralqx.operators.computational.misc.numba) U1HolonomyJax (class in neuralqx.operators.computational.misc.jax) U1HolonomyOperator() (in module neuralqx.operators.computational.misc) (in module neuralqx.operators.computational.misc.dispatch) U1InvariantPlaquetteSampler (class in neuralqx.samplers) (class in neuralqx.samplers.rules)

U1Plaquette (class in neuralqx.samplers.types) U1UnconstrainedStateEnumerator (class in neuralqx.hilbert.u1.index) (class in neuralqx.hilbert.u1.index.enumerator) UnconstrainedHilbertU1Core (class in neuralqx.hilbert.u1.unconstrained_core) UnspecifiedGaugeFixingError UnsupportedRecouplingSchemeError Update (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.dsl) (class in neuralqx.experimental.operators.symbolic.dsl.rewrite) update_parameters() (AbstractVariationalDriver method) update_program (EmissionSpec attribute), [1], [2], [3] (SymbolicIRTerm attribute), [1], [2], [3] UpdateOp (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.update) UpdateProgram (class in neuralqx.experimental.operators.symbolic), [1] (class in neuralqx.experimental.operators.symbolic.ir) (class in neuralqx.experimental.operators.symbolic.ir.update)

V

vacuum_value (NoVacuumProjector attribute) valence() (AbstractGraph method) validate_and_insert_keys() (in module neuralqx.graph.core.utils.checks) validate_symbolic_ir() (in module neuralqx.experimental.operators.symbolic.ir) (in module neuralqx.experimental.operators.symbolic.ir.validate) ValidationError variables (MultiMCState property) (VariationalState property) variational_state (MultiSolver property) VariationalMixedState (class in neuralqx.vqs) VariationalState (class in neuralqx.vqs) (class in neuralqx.vqs.base) VerificationResult (class in neuralqx.utils.misc.auth) verify_file() (Authenticator static method) verify_payload() (Authenticator static method) verify_signature() (Authenticator static method) Version (class in neuralqx.utils.module.version) (class in neuralqx.utils.module.version.version) (class in neuralqx.utils.module.version_check) version (NeuralqxVersion property) VersionInput (in module neuralqx.utils.module.version) (in module neuralqx.utils.module.version_check) vertex_valence (SingleVertexGraph property) vertices (AbstractGraph property) (GraphSymmetries property) vertices_k (AbstractLqxSphericalBoSw property) (HalfLadderGraph property) (LqxBojowaldSwiderski property) (LqxSphericalBoSw property) vertices_mu (HalfLadderGraph property) view() (AbstractHilbertSpace method) (ConstrainedHilbertU1Core method) (UnconstrainedHilbertU1Core method)

VMC (class in neuralqx.driver) (class in neuralqx.driver.vmc) vmc_driver (MultiSolver property) volume() (LqxBojowaldSwiderski method) (LqxWCL method) (LqxWCL1D method) (LqxWCL3D method) (LqxWCL4D method) (LqxWclModel method) volume_cost_kernel() (in module neuralqx.vqs.mc.kernels) VolumeOperator (class in neuralqx.operators.computational.Euclidean3d.numba) (class in neuralqx.operators.computational.Euclidean3d.numba.volume) (class in neuralqx.operators.computational.Euclidean4d.numba) (class in neuralqx.operators.computational.Euclidean4d.numba.volume) VolumeOperator() (in module neuralqx.operators.computational.Euclidean3d) (in module neuralqx.operators.computational.Euclidean3d.dispatch) (in module neuralqx.operators.computational.Euclidean4d) (in module neuralqx.operators.computational.Euclidean4d.dispatch) VolumeOperatorJax (class in neuralqx.operators.computational.Euclidean3d.jax) (class in neuralqx.operators.computational.Euclidean3d.jax.volume) (class in neuralqx.operators.computational.Euclidean4d.jax) (class in neuralqx.operators.computational.Euclidean4d.jax.volume) VolumeOperatorJaxSqrt (class in neuralqx.operators.computational.Euclidean3d.jax) (class in neuralqx.operators.computational.Euclidean3d.jax.volume) VolumeOperatorJaxSquared (class in neuralqx.operators.computational.Euclidean3d.jax) (class in neuralqx.operators.computational.Euclidean3d.jax.volume) VolumeOperatorSqrt (class in neuralqx.operators.computational.Euclidean3d.numba) (class in neuralqx.operators.computational.Euclidean3d.numba.volume) VolumeOperatorSquared (class in neuralqx.operators.computational.Euclidean3d.numba) (class in neuralqx.operators.computational.Euclidean3d.numba.volume)

W

Weighted (class in neuralqx.samplers.types) WeightedSamplerRule (class in neuralqx.samplers.rules) (in module neuralqx.samplers) wrap_callable() (in module neuralqx.profile) (in module neuralqx.profile.decorators) wrap_mod() (in module neuralqx.operators.computational.Euclidean4d.jax.gauss) wrap_model() (in module neuralqx.nn.projectors.group_projector) (in module neuralqx.nn.projectors.group_projector.group_projector) (in module neuralqx.nn.projectors.vacuum) (in module neuralqx.nn.projectors.vacuum.novacuum_projector)

wrap_trunk_as_stmh() (in module neuralqx.experimental.nn.projectors.stmh) write() (in module neuralqx.experimental.operators.symbolic), [1] (in module neuralqx.experimental.operators.symbolic.dsl) (in module neuralqx.experimental.operators.symbolic.dsl.rewrite) (Update method), [1], [2], [3] write_log_to_file() (Logger method)

Y

Yogi (class in neuralqx.optimizer.types)

Z

zeros_complex() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime) zeros_index() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime)

zeros_real() (in module neuralqx.utils.dtypes) (in module neuralqx.utils.dtypes.runtime)