Skip to content

jaxfne v0.3.41

Latest

Choose a tag to compare

@HNXJ HNXJ released this 14 Jun 19:02
· 9 commits to main since this release

JAX-optimized kernels and a public-documentation completeness pass.

Highlights

  • JAX-optimized kernels (+9 public APIs, __all__=194): spectral readouts (spectrolaminar_psd_jax, bandpower_jax, spectrolaminar_*_kernel_jax), compile_connection_rules_jax, STDP (update_stdp_weights_jax, run_stdp_stream), and an experimental volume-conductor placeholder.
  • Device-backend enforcement via jax.default_device (CPU/GPU/TPU); per-RuntimeConfig dtype (float32/float64) and jit remain adjustable. Cached jit with the recompilation guard (N_compile <= 1).
  • jit portability fix: @partial(jax.jit, ...) (decorator-factory form removed).
  • Public-doc completeness pass: scope boundaries and reserved field regimes centralized on a single page; neutral repository naming.

Truth gates (unchanged)

Proxy readouts only: field_solver_status=laminar_proxy_no_pde, physical_amplitude_claim_allowed=False. jaxfne is a computational scaffold; field/EEG/MEG/LFP/CSD outputs are proxies in relative units.

Install

pip install jaxfne==0.3.41

PyPI: https://pypi.org/project/jaxfne/0.3.41/