Skip to main content

API reference

This section documents the public surface of each package — the symbols exported from each package's top-level __init__. It is organized to match the workspace tiers.

Stable workspace

The numerically load-bearing core, held to the strictest gates.

PackageImport rootPage
Pure-Python mathomnibias.corecore
PyTorch backendomnibias.torchtorch
JAX backendomnibias.jaxjax
FermiNet / VMC bridgeomnibias.ferminetferminet

Scientific extensions

Higher-level packages built on the substrate. See the stability matrix for maturity.

PackageImport rootPage
Physics-informed NNsomnibias.pinnpinn
Field substrateomnibias.fieldsfields
Differential geometryomnibias.geometrygeometry
Equation discoveryomnibias.symbolicsymbolic
Certified verificationomnibias.verifyverify
Validated dynamicsomnibias.dynamicsdynamics

Import conventions

  • Backends are separate distributions. Install the one you need (pip install omnibias-torch or omnibias-jax); omnibias-core alone has no backend.
  • Backend subpackages are not imported eagerly. For example, importing omnibias.pinn does not pull in torch or jax; you import omnibias.pinn.torch / omnibias.pinn.jax explicitly.
  • Lazy heavy imports. Importing omnibias.ferminet does not import JAX or a FermiNet checkout; the integration modules import JAX on demand.

Versioning

Each package carries a __version__. The stable workspace currently sits at omnibias.core/torch/jax 0.4.0, ferminet 0.2.0; extensions version independently. These docs are versioning-ready — see the changelog.

Source of truth

The authoritative list of public symbols is each package's __all__. These pages summarize and group those symbols; when in doubt, the package export wins.