jaxcont

JaxCont

Differentiable continuation and bifurcation analysis in JAX.

JaxCont turns an equilibrium continuation sweep into a JAX program. Follow branches through folds with pseudo-arclength continuation, detect and refine fold/Hopf events, and compose the analysis with jax.jit, jax.vmap, and automatic differentiation.

Why JaxCont?

Transform the whole analysis. Batch ensembles of branches with vmap and differentiate bifurcation locations for inverse design.
Stay on the branch. Adaptive pseudo-arclength continuation passes folds; natural continuation is available when the parameter remains regular.
See and validate the result. Plot stability-aware bifurcation diagrams and eigenvalue trajectories with JaxCont events plus MatCont, BifurcationKit, or analytic reference markers.

The supported v0.1 surface focuses deliberately on equilibria. Periodic orbits, branch switching, and two-parameter continuation remain future work.

Installation

JaxCont requires Python 3.9 or newer.

pip install jaxcont

See the JAX installation guide when selecting CPU, GPU, or TPU support.

Quick start

Continue u² + p = 0 through its fold at p = 0:

import jax.numpy as jnp
import matplotlib.pyplot as plt
import jaxcont as jc

def saddle_node(u, p, args):
    return u**2 + p

problem = jc.bif_problem(
    saddle_node,
    u0=jnp.array([1.0]),
    p0=-1.0,
    state_names=["u"],
    param_name="p",
)

result = jc.continuation(
    problem,
    p_span=(-1.0, 0.2),
    settings=jc.ContinuationPar(ds=0.03, max_steps=200),
    events=[jc.Fold()],
)

print([(event.kind, event.p) for event in result.events])
result.plot(annotate=True, title="Saddle-node bifurcation")
plt.show()

The same result can drive a stability-spectrum plot:

from jaxcont.viz import plot_eigenvalues

plot_eigenvalues(result, shade_stability=True)

Explore

Quickstart
Example gallery
example_03_van_der_pol.py — Hopf crossing with JaxCont and MatCont markers
example_06_vmap_sweep.py — batched continuation with vmap
example_07_differentiable.py — differentiable bifurcation analysis

Development

git clone https://github.com/Ziaeemehr/JaxCont.git
cd JaxCont
python -m pip install -e ".[dev]"
python -m pytest

Contributions are welcome; see CONTRIBUTING.md, the roadmap, and the cross-validation plan.

Citation

If JaxCont supports your research, cite the archived release using its GitHub/Zenodo DOI. Citation metadata is provided in CITATION.cff. Until the first archive is minted:

@software{ziaeemehr_jaxcont_2026,
  author  = {Ziaeemehr, Abolfazl},
  title   = {JaxCont: Differentiable Continuation and Bifurcation Analysis in JAX},
  year    = {2026},
  version = {0.1.0},
  url     = {https://github.com/Ziaeemehr/JaxCont}
}

JaxCont is distributed under the MIT License.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

jaxcont-0.2.0.tar.gz (80.7 kB view details)

Uploaded Jul 24, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

jaxcont-0.2.0-py3-none-any.whl (54.3 kB view details)

Uploaded Jul 24, 2026 Python 3

File details

Details for the file jaxcont-0.2.0.tar.gz.

File metadata

Download URL: jaxcont-0.2.0.tar.gz
Upload date: Jul 24, 2026
Size: 80.7 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for jaxcont-0.2.0.tar.gz
Algorithm	Hash digest
SHA256	`64c70a73546b25a85ecc5ae8553da8d889719f3be45290d7a3c2c4577ec388a0`
MD5	`5fdffb6c7b2c911f9293c070301bb170`
BLAKE2b-256	`c2bc1c99e36a10bf08cedb84d1e8359d3540a88b9566d11ee63a9af45115d1c6`

See more details on using hashes here.

Provenance

The following attestation bundles were made for jaxcont-0.2.0.tar.gz:

Publisher: publish.yml on Ziaeemehr/JaxCont

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: jaxcont-0.2.0.tar.gz
- Subject digest: 64c70a73546b25a85ecc5ae8553da8d889719f3be45290d7a3c2c4577ec388a0
- Sigstore transparency entry: 2235870653
- Sigstore integration time: Jul 24, 2026
Source repository:
- Permalink: Ziaeemehr/JaxCont@d3424521b49bde985cab4ee7871d25697fbebf75
- Branch / Tag: refs/heads/main
- Owner: https://github.com/Ziaeemehr
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish.yml@d3424521b49bde985cab4ee7871d25697fbebf75
- Trigger Event: workflow_dispatch