The StringJAX ecosystem

The StringJAX ecosystem#

StringJAX is a project and brand, not a single physics package. The code lives in independent, separately versioned, separately citable repositories that interoperate through well-defined, version-pinned interfaces.

This page is the orientation map. Each member package has its own authoritative documentation; the descriptions below link out to those.

Public member packages#

jaxvacua — Type IIB flux-vacuum engine

The package described in the StringJAX release paper. Covers periods and special geometry; the Kähler potential, Kähler metric, and gauge-kinetic matrix; the flux superpotential and its covariant derivatives; vacuum finding through ISD sampling, linearised ISD optimisation, and Newton-type solvers; flux bounding; conifold and coni-LCS limits; and reduced (freezer) effective theories.

pip install jaxvacua  ·  GitHub  ·  docs (when public)

https://jaxvacua.readthedocs.io
jaxpolylog — Differentiable polylogarithms

JAX-compatible polylogarithm functions with automatic differentiation and JIT support, used by jaxvacua to resum worldsheet-instanton contributions to the prepotential. Designed to stay numerically stable through arbitrary-order automatic differentiation, including in deep large-complex-structure regimes where \(|z|\) may be as small as \(10^{-30}\).

pip install jaxpolylog  ·  GitHub  ·  docs

https://jaxpolylog.readthedocs.io
stringforge — Databases and vacua vault

The data layer: curated Calabi–Yau geometry databases (TDF, CICY, KKLT) served from HuggingFace and cached locally, together with the persistent vault of flux vacua. Owns I/O, caching, mirror-convention bridging, model loading, and vault layout / validation / curation; remains solver-light.

pip install stringforge  ·  GitHub  ·  docs (when public)

https://github.com/AndreasSchachner/stringforge

How they fit together#

        flowchart LR
    subgraph data[Data layer]
        SF[stringforge<br/>CY databases + vacua vault]
    end
    subgraph compute[Compute layer]
        JV[jaxvacua<br/>flux-vacuum engine]
        JP[jaxpolylog<br/>differentiable polylogs]
    end
    DATASETS[(HuggingFace<br/>cy-database)]
    SF --> JV
    JP --> JV
    DATASETS --> SF
    JV --> RESULTS[Vacua + observables]
    RESULTS --> SF

In words:

  • stringforge hosts the data the rest of the ecosystem consumes – Calabi–Yau geometries plus a persistent vault of flux vacua – and exposes them as cached, lazily-loaded objects.

  • jaxpolylog is a pure-leaf dependency. It provides the polylogarithm primitive that jaxvacua calls to evaluate worldsheet-instanton corrections.

  • jaxvacua is the physics engine. It consumes geometries from stringforge and the polylog primitive from jaxpolylog, constructs the flux effective theory, solves for vacua, and writes the result back to the stringforge vault for archival and reuse.

The umbrella stringjax package itself does not introduce any physics or new APIs. It pins the compatibility window of the three member packages and ships a thin command-line tool (stringjax doctor, stringjax versions) for environment checks.

Member-package boundaries#

Physics computations

Period vectors, prepotentials, Kähler structures, flux \(F\)-terms, scalar potentials, Hessians, mass matrices, ISD samplers, conifold limits, freezer EFTs.

Owned by jaxvacua.

Special functions

JAX-compatible polylogarithms with autodiff.

Owned by jaxpolylog.

Data movement

HuggingFace download / cache, mirror-convention bridging, model loading, vault layout, validation, curation.

Owned by stringforge.

Compatibility

Version pins, command-line environment checks, getting-started glue, this documentation hub.

Owned by stringjax.

The boundaries are deliberate. Each member package can evolve at its own pace, be released with its own DOI, and be cited independently. StringJAX guarantees that a recent combination of versions tested together is the easy default.

Planned siblings#

Future member packages are planned but not yet public. They will be released with their own repositories, papers, and documentation sites, and added to the ecosystem with their own compatibility windows.

  • Kähler-sector extensions – moduli stabilisation for the Kähler sector of general four-dimensional \(\mathcal{N}=1\) effective theories.

  • Axion-sector extensions – spectra, decay constants, and couplings for multi-axion effective theories.

The shape of the ecosystem (a thin umbrella, multiple physics packages, a shared data layer, a leaf special-functions package) is chosen so that adding such siblings is a routine extension rather than a re-architecture.