Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
138 commits
Select commit Hold shift + click to select a range
d90d0e7
shift VectorFieldNet type from estimators/flowmatching_estimator.py t…
jaivardhankapoor Mar 17, 2025
1e77a4a
refactor: update imports and enhance ConditionalScoreEstimator
jaivardhankapoor Mar 18, 2025
3b380cf
Update z-score parameters in flowmatching and posterior score neural …
jaivardhankapoor Mar 18, 2025
1ed9d26
refactor: update vector field neural network architecture
jaivardhankapoor Mar 19, 2025
b878315
update: fix handling of shapes during sampling for npse and fmpe
jaivardhankapoor Mar 19, 2025
2121dc1
refactor: improve docstring formatting and remove commented-out code
jaivardhankapoor Mar 20, 2025
6a36d3c
refactor: integrate embedding net in flowmatching estimator
jaivardhankapoor Mar 20, 2025
59ee524
refactor: update flowmatching estimator and vector field network init…
jaivardhankapoor Mar 21, 2025
e0398c5
Merge commit '59ee524d9400331f0b0e359221d2445f8cc25d64' into merge_fl…
manuelgloeckler Mar 28, 2025
8217315
Fix some problems
manuelgloeckler Mar 28, 2025
c77970e
To run CI, comment out broken tests
manuelgloeckler Mar 28, 2025
bc32811
Updates: Internal nets should be shared, but Estimator builders shoul…
manuelgloeckler Apr 3, 2025
c51ecad
Unify shape handling in score and flow.
manuelgloeckler Apr 3, 2025
4c7277a
Formating to get CI going (failing tests expected)
manuelgloeckler Apr 3, 2025
b334c92
Merge remote-tracking branch 'origin/main' into merge_flow_builders_t…
manuelgloeckler May 15, 2025
8db0e6c
Some small fixes and refactorings
manuelgloeckler May 15, 2025
9bcc536
Fix ruff things
manuelgloeckler May 15, 2025
342bc6b
Fixing score sampler tests with new net builder API
manuelgloeckler May 15, 2025
fc19720
Fixing flow estimator bugs
manuelgloeckler May 15, 2025
8d4b2de
Bug hunting + fixing
manuelgloeckler May 15, 2025
c0afa45
Rearrange trainers + fixing tests to not use "special" hyperparameter…
manuelgloeckler May 15, 2025
cc5b3e3
Fix ruff
manuelgloeckler May 15, 2025
e7c6d5b
Fixing failing tests
manuelgloeckler May 15, 2025
5000bd5
Fix validation loss check
manuelgloeckler May 15, 2025
5ad2c5d
Fix for new FMPE args
manuelgloeckler May 15, 2025
2c47282
Consistent naming for FMPE and NPSE
manuelgloeckler May 21, 2025
beba5fc
Bug fix for Neural ODE sampling with ScoreEstimators
manuelgloeckler May 21, 2025
2a3a6a2
less numsims for vfestimators in tests
manuelgloeckler May 21, 2025
f8ef3d3
Test changes
manuelgloeckler May 21, 2025
d002a70
Merge remote-tracking branch 'origin/main' into merge_flow_builders_t…
manuelgloeckler May 21, 2025
a22e38b
Bug fix, bounded epochs on default??? Add a better convergence chechk...
manuelgloeckler May 27, 2025
1f125ab
Remove print
manuelgloeckler May 27, 2025
0378783
new mlp which performs better...
manuelgloeckler May 27, 2025
7b56a52
Allow setting num_sims in minisbibm for eval
manuelgloeckler May 27, 2025
1619b82
Add arg for num sims, cache results by default
manuelgloeckler May 27, 2025
621255d
some refactorings
manuelgloeckler May 27, 2025
d2c7ce4
Fix formating
manuelgloeckler May 28, 2025
eafd18d
Formatting, make defaults more uniform
manuelgloeckler May 28, 2025
38fc015
Make factories more SBI-like
manuelgloeckler May 28, 2025
4e6f9e8
Some estimator tests added
manuelgloeckler May 28, 2025
87f17e0
Formatting, fix kwargs errors, more tests
manuelgloeckler May 28, 2025
18337d4
Fix tests and init transformer last layer as zero
manuelgloeckler May 28, 2025
c33cd92
Remove test jupyter :/
manuelgloeckler May 28, 2025
a9ded55
Formatting, refactoring tests
manuelgloeckler May 28, 2025
1331d78
Fix pyright
manuelgloeckler May 28, 2025
e91a9f0
Remove what is expected to fail
manuelgloeckler May 28, 2025
46f86c2
Minor fixes
manuelgloeckler Jul 2, 2025
3f3fcc5
Small docstring update
manuelgloeckler Jul 3, 2025
380b334
Backward compatiblity warnings from some unused kwargs
manuelgloeckler Jul 3, 2025
14262f1
Typing with vectorfield net
manuelgloeckler Jul 3, 2025
cf762ea
Simplify score estimator
manuelgloeckler Jul 3, 2025
2839aec
Updates
manuelgloeckler Jul 3, 2025
151973a
Fixing transformer with cross attn
manuelgloeckler Jul 9, 2025
b23418b
Add error msg for unsupported shapes
manuelgloeckler Jul 9, 2025
689a727
Better tests
manuelgloeckler Jul 9, 2025
0b3dc40
refactored tests
manuelgloeckler Jul 9, 2025
49f8caf
Reverting wierd reshapings in score estimator.
manuelgloeckler Jul 9, 2025
66a68eb
Merge branch 'main' into merge_flow_builders_to_current_main
manuelgloeckler Jul 9, 2025
0293383
Fix formating issues
manuelgloeckler Jul 9, 2025
c585eb1
Fixing inconsistencies
manuelgloeckler Jul 9, 2025
42c1d56
Fixing pyright
manuelgloeckler Jul 9, 2025
024f54e
Fix embedding_net not passed
manuelgloeckler Jul 9, 2025
59e35a0
Fix embedding net bug
manuelgloeckler Jul 9, 2025
f1cf710
Remove redundant "num_blocks"
manuelgloeckler Jul 9, 2025
bed97b6
Merge branch 'main' into merge_flow_builders_to_current_main
manuelgloeckler Jul 11, 2025
a0fb73d
Adding some degree of backward compatibility on user interface.
manuelgloeckler Jul 11, 2025
387f965
Fixing failing test on new convergence check
manuelgloeckler Jul 11, 2025
fe00c02
Add transformer to bm
manuelgloeckler Jul 15, 2025
1c00bcd
Must be okay that the files already exits bm
manuelgloeckler Jul 15, 2025
27a9c3a
Fix merge bug. Add deprecation warnings for Score estimator keyword a…
manuelgloeckler Jul 15, 2025
a9445ab
Fixing transformers... (no pos emb. and others)
manuelgloeckler Jul 15, 2025
0dca35b
Refactorings and tunings
manuelgloeckler Jul 15, 2025
25af448
deprecation warnings and small refactorings
manuelgloeckler Jul 15, 2025
2eef0c6
Backwards compatibility
manuelgloeckler Jul 15, 2025
587eda3
Move score_estimator tests to vf_estimator_tests, run doc notebook once
manuelgloeckler Jul 15, 2025
af6acd1
remove random wierd comment
manuelgloeckler Jul 15, 2025
4e15f02
Remove tolerance special cases
manuelgloeckler Jul 15, 2025
27dda74
Consistent naming
manuelgloeckler Jul 15, 2025
ed1902c
Faster convergence for slighly worse performance
manuelgloeckler Jul 16, 2025
870a754
Backward compatibility for imports of NPSE and FMPE
manuelgloeckler Jul 16, 2025
8e2ed99
Docstring update
manuelgloeckler Jul 16, 2025
1e073e3
Imporve docstrings
manuelgloeckler Jul 16, 2025
1fb0664
Backward compatibility
manuelgloeckler Jul 16, 2025
0a04880
Use new keywords
manuelgloeckler Jul 16, 2025
4156549
Merge branch 'main' into merge_flow_builders_to_current_main
manuelgloeckler Jul 16, 2025
0679c22
Merge branch 'main' into merge_flow_builders_to_current_main
manuelgloeckler Jul 23, 2025
342da44
Format
manuelgloeckler Jul 23, 2025
e9e6f64
Add missing headers
manuelgloeckler Jul 24, 2025
4c483b1
Update sbi/inference/trainers/vfpe/base_vf_inference.py
manuelgloeckler Aug 7, 2025
7e272b8
Update sbi/inference/trainers/fmpe/__init__.py
manuelgloeckler Aug 7, 2025
6b99778
Update sbi/inference/trainers/npse/__init__.py
manuelgloeckler Aug 7, 2025
427f3c0
Update sbi/inference/trainers/vfpe/fmpe.py
manuelgloeckler Aug 7, 2025
5d03a84
Update sbi/inference/trainers/vfpe/fmpe.py
manuelgloeckler Aug 7, 2025
a5e216e
Update sbi/inference/trainers/vfpe/npse.py
manuelgloeckler Aug 7, 2025
f4ca1f8
Update sbi/inference/trainers/vfpe/npse.py
manuelgloeckler Aug 7, 2025
0a09f67
Update sbi/neural_nets/__init__.py
manuelgloeckler Aug 7, 2025
099ad7d
Update sbi/neural_nets/estimators/flowmatching_estimator.py
manuelgloeckler Aug 7, 2025
663088e
Update sbi/neural_nets/estimators/score_estimator.py
manuelgloeckler Aug 7, 2025
43f21be
Update sbi/neural_nets/factory.py
manuelgloeckler Aug 7, 2025
dfd51e4
Update sbi/neural_nets/factory.py
manuelgloeckler Aug 7, 2025
c2bd2d3
Update sbi/neural_nets/net_builders/vector_field_nets.py
manuelgloeckler Aug 7, 2025
8edbfaa
Update sbi/neural_nets/net_builders/vector_field_nets.py
manuelgloeckler Aug 7, 2025
846debb
Update sbi/neural_nets/factory.py
manuelgloeckler Aug 7, 2025
09cd09e
Update tests/bm_test.py
manuelgloeckler Aug 7, 2025
8c613ce
Update tests/bm_test.py
manuelgloeckler Aug 7, 2025
f8a8b50
Update tests/bm_test.py
manuelgloeckler Aug 7, 2025
9276a78
Update tests/bm_test.py
manuelgloeckler Aug 7, 2025
cc97264
Merge branch 'main' into merge_flow_builders_to_current_main
manuelgloeckler Aug 7, 2025
457a0aa
Merge remote-tracking branch 'origin/main' into merge_flow_builders_t…
manuelgloeckler Aug 7, 2025
fab3c62
Add nugget as keyward argument to train
manuelgloeckler Aug 7, 2025
a1d9b98
Imporve converged docstring
manuelgloeckler Aug 7, 2025
1ced05a
Better typing and docstrings and so on
manuelgloeckler Aug 7, 2025
34f396f
docstring
manuelgloeckler Aug 7, 2025
4b09869
Add context
manuelgloeckler Aug 7, 2025
06b84a5
Extended docstring
manuelgloeckler Aug 7, 2025
7eeaa7d
move protocol
manuelgloeckler Aug 7, 2025
57fdf39
Formating fix
manuelgloeckler Aug 7, 2025
f661a05
Revert "move protocol"
manuelgloeckler Aug 7, 2025
50f7443
fix formating
manuelgloeckler Aug 7, 2025
5cb49b7
removing deprecated
manuelgloeckler Aug 7, 2025
3caf37a
Fix typing
manuelgloeckler Aug 7, 2025
4ea5e3a
Positional argument for default builder model name
manuelgloeckler Aug 8, 2025
7f281cd
Formating
manuelgloeckler Aug 8, 2025
4f88303
unify nets test
manuelgloeckler Aug 8, 2025
1691a63
fixing builder
manuelgloeckler Aug 8, 2025
77ba7a6
update notebooks
manuelgloeckler Aug 8, 2025
10f61b6
Formating and some text updates
manuelgloeckler Aug 8, 2025
314f45e
formating
manuelgloeckler Aug 8, 2025
77b4139
fix deprecation warning on default args
janfb Aug 13, 2025
b07d188
unnecessary
manuelgloeckler Aug 14, 2025
ebb8706
remove unecessary notes
manuelgloeckler Aug 14, 2025
d449e46
refactor check for deprecation warning
janfb Aug 14, 2025
109f6be
fix mcmc params passing in test
janfb Aug 14, 2025
fca9e5a
Fix mnle_test
manuelgloeckler Aug 14, 2025
b646608
Merge branch 'main' into merge_flow_builders_to_current_main
janfb Aug 14, 2025
d1c1346
add missing import
janfb Aug 14, 2025
b9f35fb
Notebooks rerun without warning and with striped notebook outputs
manuelgloeckler Aug 14, 2025
0919206
Merge remote-tracking branch 'origin/main' into fix_docs
manuelgloeckler Aug 15, 2025
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
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -164,13 +164,13 @@ how to run each of these methods
and hassle-free simulation-based inference_](https://arxiv.org/abs/2210.04815)
(NeurIPS 2022).

- [`FMPE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.fmpe.fmpe.FMPE)
- [`FMPE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.vfpe.fmpe.FMPE)
from Wildberger, J., Dax, M., Buchholz, S., Green, S., Macke, J. H., & Schölkopf, B.
[_Flow matching for scalable simulation-based
inference_](https://proceedings.neurips.cc/paper_files/paper/2023/hash/3663ae53ec078860bb0b9c6606e092a0-Abstract-Conference.html).
(NeurIPS 2023).

- [`NPSE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.npse.npse.NPSE) from
- [`NPSE`](https://sbi-dev.github.io/sbi/latest/reference/#sbi.inference.trainers.vfpe.npse.NPSE) from
Geffner, T., Papamakarios, G., & Mnih, A. [_Compositional score modeling for
simulation-based inference_](https://proceedings.mlr.press/v202/geffner23a.html).
(ICML 2023)
Expand Down
123 changes: 61 additions & 62 deletions docs/tutorials/01_Bayesian_workflow.ipynb

Large diffs are not rendered by default.

4 changes: 2 additions & 2 deletions mkdocs/docs/reference/inference.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,12 +17,12 @@
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true

::: sbi.inference.trainers.fmpe.fmpe.FMPE
::: sbi.inference.trainers.vfpe.fmpe.FMPE
selection:
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true

::: sbi.inference.trainers.npse.npse.NPSE
::: sbi.inference.trainers.vfpe.npse.NPSE
selection:
filters: [ "!^_", "^__", "!^__class__" ]
inherited_members: true
Expand Down