Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
b6292b1
Add design plan for the notebook / Colab API
cweniger May 17, 2026
1c807f1
Merge branch 'main' into plan/colab-notebook-api
cweniger Jun 8, 2026
01dcf7b
Update COLAB_API_PLAN: defer Step 0, simplify API surface
cweniger Jun 8, 2026
a21f0b1
Step 1: extract _run_pipeline from launch_mode
cweniger Jun 8, 2026
a28a425
Step 2: cloudpickle spike — all scenarios pass
cweniger Jun 8, 2026
be59a43
Step 3: flat config surface for Flow; GaussianPosterior privatised
cweniger Jun 8, 2026
bbaf70e
Add DynamicSVD and ToeplitzWhitener to falcon.embeddings
cweniger Jun 8, 2026
c6cf3c7
Step 4: falcon.init / launch / shutdown + 01_minimal notebook
cweniger Jun 8, 2026
4cfe23f
Step 5: Graph.add_node, live-object support, escape-hatch serialization
cweniger Jun 8, 2026
506af12
Add DynamicSVD embedding mode to example 05
cweniger Jun 8, 2026
0c376ed
Step 6: rich repr for Graph and Run
cweniger Jun 8, 2026
2971058
Add notebook.ipynb for 01_minimal and 04_gaussian examples
cweniger Jun 8, 2026
eb6b192
Remove accidental asyncio wrapper from DeployedGraph._launch
cweniger Jun 8, 2026
79247ba
Remove fixed name from DatasetManagerActor
cweniger Jun 8, 2026
c766a68
Fix _prepare_config mutating the caller's Config object
cweniger Jun 8, 2026
f826b2b
Fix None embedding crash in GaussianFullCov._create_model
cweniger Jun 8, 2026
5456785
Fix Graph-based launch crash when output dir has stale config.yml
cweniger Jun 8, 2026
c3141cb
Refactor estimators to two-phase init/setup pattern
cweniger Jun 8, 2026
518dce1
Drop prefixed params: estimators now use natural __init__ names
cweniger Jun 8, 2026
e423596
Remove notebook.py files; update 04_gaussian notebook to match config…
cweniger Jun 8, 2026
c47cbc8
Merge branch 'plan/colab-notebook-api' into feature/dynamic-svd
cweniger Jun 8, 2026
f4d94e2
WIP: 05_linear_regression with DynamicSVD embedding (debugging wide p…
cweniger Jun 9, 2026
7485c1a
Fix GaussianFullCov proposal sampling using stale/None DynamicSVD com…
cweniger Jun 9, 2026
3999324
Tune 05_linear_regression config: more epochs, less snapshotting
cweniger Jun 9, 2026
1381cfa
Fix embedding sync: copy DynamicSVD state at each best-model checkpoint
cweniger Jun 9, 2026
d274630
Clean up .gitignore: replace stale entries with **/output/
cweniger Jun 9, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 1 addition & 4 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -108,10 +108,7 @@ runs/
**/wandb/

# Falcon-specific output directories
**/default-store-dir/
**/default-graph-dir/
**/outputs/
**/*.zarr/
**/output/

# Data files
#*.npz
Expand Down
28 changes: 12 additions & 16 deletions examples/01_minimal/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -76,25 +76,21 @@ graph:

estimator: # Posterior estimator network
_target_: falcon.estimators.Flow # Flow-based posterior estimation
loop: # Training loop parameters
max_epochs: 300
batch_size: 128
early_stop_patience: 32 # Early stopping patience
network: # Neural network architecture
net_type: nsf # Neural spline flow (alternatives: zuko_gf, maf, naf, etc.)
theta_norm: true # Normalize parameter space
norm_momentum: 0.003 # Momentum for online normalization updates
max_epochs: 300
net_type: nsf # Neural spline flow (alternatives: zuko_gf, maf, naf, etc.)
lr: 0.01
gamma: 0.5 # Mixing coefficient for amortization weighting
embedding: # Neural embedding for observation x
_target_: model.E
_input_: [x]
optimizer: # Optimizer parameters
lr: 0.01
lr_decay_factor: 0.5 # LR decay multiplier
scheduler_patience: 16 # LR decay after N stagnant epochs
inference: # Inference and sampling parameters
gamma: 0.5 # Mixing coefficient for amortization weighting
discard_samples: false
log_ratio_threshold: -20 # Stability cutoff for log ratios
batch_size: 128
early_stop_patience: 32
theta_norm: true # Normalize parameter space
norm_momentum: 0.003 # Momentum for online normalization updates
lr_decay_factor: 0.5 # LR decay multiplier
lr_patience: 16 # LR decay after N stagnant epochs
discard_samples: false
log_ratio_threshold: -20 # Stability cutoff for log ratios

ray:
num_gpus: 0 # GPU count per Ray worker (0 = CPU)
Expand Down
Loading
Loading