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-RuntimeConfigdtype (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