Add example 06: spectral analysis with chirp signal inference#42

Open

cweniger wants to merge 35 commits into

mainfrom

feat/06_example

Commits on Feb 22, 2026

Add example 06: spectral analysis with fuge EMRI embeddings

cweniger
and
claude
committed
Widen prior ranges and increase buffer/batch sizes for spectral example

cweniger
and
claude
committed
Fix TokenEmbed to compute z-score normalization on first batch

cweniger
and
claude
committed
Force JAX CPU backend for EMRI simulator in Ray workers

cweniger
and
claude
committed
Only fall back to JAX CPU when no GPU is visible

cweniger
and
claude
committed
Add Fisher/CRB comparison plotting script for spectral example

cweniger
and
claude
committed
Default make_plots.py to outputs/latest

cweniger
and
claude
committed
Add chunk_size to avoid OOM during posterior sampling

cweniger
and
claude
committed
Tune spectral analysis buffer and sampling parameters

cweniger
and
claude
committed
Detach STFT output to avoid wasteful backward pass and add spectra animation script

cweniger
and
claude
committed
Support deterministic intermediate nodes in inference graph and add tokens caching config

cweniger
and
claude
committed
Mixed-precision dtype zones and explicit online normalization

cweniger
and
claude
committed
Add dim parameter to RunningNorm in example 06 configs

cweniger
and
claude
committed
Trigger CI
cweniger
committed

Commits on Feb 25, 2026

GPU-accelerate signal generation with JAX vmap, scale to N=1M

cweniger
and
claude
committed

Commits on Mar 3, 2026

Consolidate example 06 benchmark: local chirp model, 10σ CRB priors, N=100k

cweniger
and
claude
committed

Commits on Mar 4, 2026

Fix float64 precision for GaussianPosterior output buffers

cweniger
and
claude
committed

Commits on Mar 6, 2026

Commits on Mar 7, 2026

Add SpectralTokenEmbed with Fourier frequency features and PhaseFormer design doc

cweniger
and
claude
committed

Commits on Mar 9, 2026

Commits on Mar 10, 2026