nkdsl.compiler.defaults

Default pipeline and registry factories for the symbolic compiler.

Functions

build_default_symbolic_operator_lowering_registry()	Builds a registry with the default NetKet discrete JAX target.
default_symbolic_artifact_store()	Returns the module-level shared in-memory artifact store.
default_symbolic_lowerer_registry(*[, ...])	Builds the default symbolic lowerer registry.
default_symbolic_operator_lowering_registry()	Returns the module-level shared operator-lowering target registry.
default_symbolic_pass_pipeline()	Builds the default two-stage symbolic compiler pass pipeline.
reset_default_symbolic_singletons()	Resets module-level default singleton instances.

Classes

InMemorySymbolicArtifactStore([max_entries])	Thread-safe in-process artifact store backed by a Python dict.
JAXSymbolicLowerer(*[, ...])	JAX-backend symbolic operator lowerer.
SymbolicDiagnosticsPass()	Runs DSL lint/diagnostics checks before lowering.
SymbolicFusionPass()	Groups IR terms into fusion-compatible clusters.
SymbolicLowererRegistry([lowerers])	Registry that maps compilation contexts to concrete lowerers.
SymbolicMaxConnSizeAnalysisPass()	Computes per-term and total max-connection-size bounds.
SymbolicNormalizationPass()	Normalizes and fingerprints the symbolic operator IR.
SymbolicOperatorLoweringRegistry([targets, ...])	Registry for selectable compiled-operator targets.
SymbolicPassPipeline(*, pre_cache_passes, ...)	Staged symbolic compiler-pass pipeline.
SymbolicValidationPass()	Validates symbolic operator IR before lowering.