jaxfne

Tensor-Field Neural Equations (TFNE) expressed as a JAX simulation engine for layer-resolved neural circuits, source operators, field proxies, probes, objectives, and evidence receipts.

Scientific grammar: Emitter → Source → Field → Probe → Objective → Optimizer → Manifest

Execution grammar: CircuitSpec → construct → Model → simulate → Signals

CircuitSpec is a conceptual category (Configuration | NeuronalTensor), not a concrete public production class — construct accepts either a Configuration (original path) or a NeuronalTensor with a RuntimeConfiguration (tensor-first path). It is unrelated to the experimental jaxfne.experimental_hpc.CircuitSpec type, which is not accepted by production construct.

Adaptation (optional HDP family): finite-dimensional hidden biophysical state $H$ and adaptive parameter coordinates $\Theta$ (synaptic and intrinsic), mediated by

$$ \dot X = F_X(X,H,\Theta,U),\quad \dot H = F_H(H,X,\Theta,U),\quad \dot\Theta = F_\Theta(H,X,\Theta). $$

$H$-state is the latent representation; HDP is the adaptive dynamical formulation that uses it. See H-state / HDP guide.

Install

pip install jaxfne
pip install "jaxfne[viz]"   # optional plotting

Development: pip install -e ".[dev,viz]" after cloning.

Minimal example

import jaxfne as jtfne

jtfne.enable_x64()
tensor  = jtfne.load_canonical_neuronal_tensor("canonical-v1-column-1000n")
model   = jtfne.construct(tensor, jtfne.RuntimeConfiguration(seed=0, duration_ms=1000.0, dt_ms=0.5))
signals = jtfne.simulate(model)

Optional: generative development (JDNA)

Instead of loading a fixed tensor, generate one from a PseudoGenome:

genome  = jtfne.load_canonical_pseudogenome("canonical-v1-column-1000n")
tensor  = jtfne.develop(genome, seed=0)   # K_D: development seed
model   = jtfne.construct(tensor, jtfne.RuntimeConfiguration(seed=1, duration_ms=1000.0, dt_ms=0.5))
signals = jtfne.simulate(model)

JDNA is an optional path; the direct Configuration/NeuronalTensor paths remain first-class. See JDNA guide.

JaxFNE composes established neural dynamical models, source/readout operators, and JAX-native execution under the TFNE typed state/map grammar. See TFNE theory and References.

Import convention: import jaxfne as jtfne. Builder paths, paradigms, and optimization: Quickstart.

Relative and calibrated outputs

Simulated quantities are relative by default. Calibrated claims require an explicit transformation with evidence. See Scope & status.

Documentation

Resource	Link
Quickstart	docs/quickstart.md
Site	jaxfne.readthedocs.io
Tutorials	docs/tutorials/
Études	docs/etudes/
Frozen compatibility contract (0.4.13)	docs/public_surface_contract.md
Changelog	docs/changelog.md

Citation

CITATION.cff · citation guide

Visual examples

Real package outputs (simulated proxy scaffolds; calibration requires an explicit transformation with evidence): circuit geometry from Model.neuron_table(), a 10 s spike raster, a laminar spectrolaminar readout, and long-timescale firing-rate dynamics with homeostasis on/off. More: showcases.

_{Canonical laminar column — geometry from `neuron_table()`}	_{10 s spike raster — homeostasis on (proxy)}
_{Spectrolaminar proxy readout — depth-graded homeostasis}	_{Population rate over 10 s — homeostasis off vs on (r*=10 Hz)}

Release files for jaxfne 0.4.18

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for jaxfne 0.4.18
File	Size	Uploaded
jaxfne-0.4.18.tar.gz	35.5 MB	Aug 28, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for jaxfne 0.4.18
File	Interpreter	ABI	Platform	Reset
jaxfne-0.4.18-py3-none-any.whl 688.1 kB Aug 28, 2026	Python 3	none	any	Details

Total release size: 36.2 MB

Release files / jaxfne-0.4.18.tar.gz

Download URL	jaxfne-0.4.18.tar.gz
Size	35.5 MB
Tags	Source
SHA-256 checksum How to use checksums	`5ec72722051fb3740f496480ee0e23f0e6e0006b9249840f03d5cb61cd74bd9f`
BLAKE2b-256 checksum How to use checksums	`167ce63b6c7028ff61c386b4c946e62688aaa7ff4e52eb231299a752b3fb1c72`
Upload date	Aug 28, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/HNXJ/jaxfne
Publishing commit: github.com/HNXJ/jaxfne/tree/875b1a98aaac05c4fca32ec90dceaab213945b6c

Workflow

Publishing configuration: github.com/HNXJ/jaxfne/blob/875b1a98aaac05c4fca32ec90dceaab213945b6c/.github/workflows/publish.yml
Publishing logs: github.com/HNXJ/jaxfne/actions/runs/33207550981/attempts/1

Artifact

Subject: jaxfne-0.4.18.tar.gz
SHA-256 checksum: 5ec72722051fb3740f496480ee0e23f0e6e0006b9249840f03d5cb61cd74bd9f Verifying attestations

Release files / jaxfne-0.4.18-py3-none-any.whl

Download URL	jaxfne-0.4.18-py3-none-any.whl
Size	688.1 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`eb65ce5b3c57ee409715bf339689fe1dc6cbf85348c77237ff13070970acc1d4`
BLAKE2b-256 checksum How to use checksums	`27f37bb1ccaba9568ad4276c907d2015cef4f715893ae2c4688f71a872d22047`
Upload date	Aug 28, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/HNXJ/jaxfne
Publishing commit: github.com/HNXJ/jaxfne/tree/875b1a98aaac05c4fca32ec90dceaab213945b6c

Workflow

Publishing configuration: github.com/HNXJ/jaxfne/blob/875b1a98aaac05c4fca32ec90dceaab213945b6c/.github/workflows/publish.yml
Publishing logs: github.com/HNXJ/jaxfne/actions/runs/33207550981/attempts/1

Artifact

Subject: jaxfne-0.4.18-py3-none-any.whl
SHA-256 checksum: eb65ce5b3c57ee409715bf339689fe1dc6cbf85348c77237ff13070970acc1d4 Verifying attestations