-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merge remote to fork #1
Commits on Feb 21, 2024
-
* Update README.md Update citation. * Update README.md
Configuration menu - View commit details
-
Copy full SHA for bc89395 - Browse repository at this point
Copy the full SHA bc89395View commit details
Commits on Feb 28, 2024
-
Indexing the notebook showing how to reproduce the GIF. (#640)
Co-authored-by: Junpeng Lao <junpenglao@gmail.com>
Configuration menu - View commit details
-
Copy full SHA for 8a9dfbf - Browse repository at this point
Copy the full SHA 8a9dfbfView commit details
Commits on Mar 7, 2024
-
* Bump python version * update bool inverse
Configuration menu - View commit details
-
Copy full SHA for 2ccdfb0 - Browse repository at this point
Copy the full SHA 2ccdfb0View commit details
Commits on Mar 25, 2024
-
SMC: allow each mutation kernel to have different parameters. (#649)
* vmaping over parameters in base * switch from mcmc_factory to just passing in parameters * pre-commit and typing * CRU and docs improvement * pre-commit * code review updates * pre-commit * rename test
Configuration menu - View commit details
-
Copy full SHA for 3dc3809 - Browse repository at this point
Copy the full SHA 3dc3809View commit details
Commits on Mar 28, 2024
-
Migrate from deprecated
host_callback
toio_callback
(#651)* Migrate from deprecated `host_callback` to `io_callback` Co-Authored-By: George Necula <gnecula@users.noreply.github.com> * Format file * Fix bug
Configuration menu - View commit details
-
Copy full SHA for 2e25624 - Browse repository at this point
Copy the full SHA 2e25624View commit details
Commits on Mar 31, 2024
-
Fix MALA transition energy (#653)
* Fix MALA transition energy * Use a different logic.
Configuration menu - View commit details
-
Copy full SHA for f77297f - Browse repository at this point
Copy the full SHA f77297fView commit details
Commits on Apr 1, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 7cf4f9d - Browse repository at this point
Copy the full SHA 7cf4f9dView commit details
Commits on Apr 8, 2024
-
Replace iterative RNG split and carry with
jax.random.fold_in
(#656)* Replace iterative RNG split and carry with `jax.random.fold_in` * revert unintended change * file formatting * change `jax.tree_map` to `jax.tree.map` * revert unintended file * fiddle with rng_key * seed again
Configuration menu - View commit details
-
Copy full SHA for a5f7482 - Browse repository at this point
Copy the full SHA a5f7482View commit details
Commits on Apr 22, 2024
-
Removal of Algorithm classes. (#657)
* more * removing export * removal of classes, tests passing * linter * fix on test * linter * removing parametrization on test * code review updates * exporting as_top_level_api in dynamic_hmc * linter * code review update: replace imports
Configuration menu - View commit details
-
Copy full SHA for 1bc6f93 - Browse repository at this point
Copy the full SHA 1bc6f93View commit details
Commits on May 8, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 3f92393 - Browse repository at this point
Copy the full SHA 3f92393View commit details -
Configuration menu - View commit details
-
Copy full SHA for 40efb6c - Browse repository at this point
Copy the full SHA 40efb6cView commit details
Commits on May 13, 2024
-
Configuration menu - View commit details
-
Copy full SHA for af79fa4 - Browse repository at this point
Copy the full SHA af79fa4View commit details
Commits on May 16, 2024
-
Enable fitlering of AdaptationInfo (#674)
* enable AdaptationInfo filtering * revert progress_bar * fix pre-commit * fix empty sets * enable adapt info filtering for all adaptation algorithms * fix precommit /progressbar=True * change filter tuple to use tree_map
Configuration menu - View commit details
-
Copy full SHA for cd91e41 - Browse repository at this point
Copy the full SHA cd91e41View commit details
Commits on May 20, 2024
-
Update
run_inference_algorithm
to splitinitial_position
and `ini……tial_state` (#672) * UPDATE DOCSTRING * ADD STREAMING VERSION * UPDATE TESTS * ADD DOCSTRING * ADD TEST * REFACTOR RUN_INFERENCE_ALGORITHM * UPDATE DOCSTRING * Precommit * CLEAN TESTS * ADD INITIAL_POSITION * FIX TEST * RENAME O * FIX DOCSTRING * PUT EXPECTATION AFTER TRANSFORM
Configuration menu - View commit details
-
Copy full SHA for e0a7f9e - Browse repository at this point
Copy the full SHA e0a7f9eView commit details
Commits on May 25, 2024
-
* TESTS * TESTS * UPDATE DOCSTRING * ADD STREAMING VERSION * ADD PRECONDITIONING TO MCLMC * ADD PRECONDITIONING TO TUNING FOR MCLMC * UPDATE GITIGNORE * UPDATE GITIGNORE * UPDATE TESTS * UPDATE TESTS * ADD DOCSTRING * ADD TEST * STREAMING AVERAGE * ADD TEST * REFACTOR RUN_INFERENCE_ALGORITHM * UPDATE DOCSTRING * Precommit * CLEAN TESTS * GITIGNORE * PRECOMMIT CLEAN UP * ADD INITIAL_POSITION * FIX TEST * ADD TEST * REMOVE BENCHMARKS * BUG FIX * CHANGE PRECISION * CHANGE PRECISION * RENAME O * UPDATE STREAMING AVG * UPDATE PR * RENAME STD_MAT
Configuration menu - View commit details
-
Copy full SHA for 5831740 - Browse repository at this point
Copy the full SHA 5831740View commit details
Commits on May 27, 2024
-
New integrator, and add some metadata to integrators.py (#681)
* TESTS * TESTS * UPDATE DOCSTRING * ADD STREAMING VERSION * ADD PRECONDITIONING TO MCLMC * ADD PRECONDITIONING TO TUNING FOR MCLMC * UPDATE GITIGNORE * UPDATE GITIGNORE * UPDATE TESTS * UPDATE TESTS * ADD DOCSTRING * ADD TEST * STREAMING AVERAGE * ADD TEST * REFACTOR RUN_INFERENCE_ALGORITHM * UPDATE DOCSTRING * Precommit * CLEAN TESTS * GITIGNORE * PRECOMMIT CLEAN UP * FIX SPELLING, ADD OMELYAN, EXPORT COEFFICIENTS * TEMPORARILY ADD BENCHMARKS * ADD INITIAL_POSITION * FIX TEST * CLEAN UP * REMOVE BENCHMARKS * ADD TEST * REMOVE BENCHMARKS * BUG FIX * CHANGE PRECISION * CHANGE PRECISION * ADD OMELYAN TEST * RENAME O * UPDATE STREAMING AVG * UPDATE PR * RENAME STD_MAT * MERGE MAIN * REMOVE COEFFICIENT EXPORTS
Configuration menu - View commit details
-
Copy full SHA for 20666de - Browse repository at this point
Copy the full SHA 20666deView commit details -
* Minor formatting * formatting * fix test * formatting
Configuration menu - View commit details
-
Copy full SHA for 360ac3b - Browse repository at this point
Copy the full SHA 360ac3bView commit details
Commits on Jun 3, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 3fbdac6 - Browse repository at this point
Copy the full SHA 3fbdac6View commit details -
Configuration menu - View commit details
-
Copy full SHA for a3e235f - Browse repository at this point
Copy the full SHA a3e235fView commit details -
Configuration menu - View commit details
-
Copy full SHA for 83bc3a0 - Browse repository at this point
Copy the full SHA 83bc3a0View commit details
Commits on Jun 5, 2024
-
Apply function on pytree directly. (#692)
* Apply function on pytree directly. Avoiding unnecssary unpacking * Fix kwarg
Configuration menu - View commit details
-
Copy full SHA for a4408d3 - Browse repository at this point
Copy the full SHA a4408d3View commit details -
Configuration menu - View commit details
-
Copy full SHA for dd9ba03 - Browse repository at this point
Copy the full SHA dd9ba03View commit details
Commits on Jun 15, 2024
-
Enable shared mcmc parameters with tempered smc (#694)
* add parameter filtering * fix parameter split + docstring * change extend_paramss
Configuration menu - View commit details
-
Copy full SHA for 3353209 - Browse repository at this point
Copy the full SHA 3353209View commit details
Commits on Jun 20, 2024
-
Configuration menu - View commit details
-
Copy full SHA for eca35ab - Browse repository at this point
Copy the full SHA eca35abView commit details
Commits on Jun 24, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 5764a2b - Browse repository at this point
Copy the full SHA 5764a2bView commit details -
* Fix equation formatting * Clarify JAX gradient error * Fix punctuation + capitalization * Fix grammar Should not begin sentence with "i.e." in English. * Fix math formatting error * Fix typo Change parallel _ensample_ chain adaptation to parallel _ensemble_ chain adaptation. * Add SVGD citation to appear in doc Currently the SVGD paper is only cited in the `kernel` function, which is defined _within_ the `build_kernel` function. Because of this nested function format, the SVGD paper is _not_ cited in the documentation. To fix this, I added a citation to the SVGD paper in the `as_top_level_api` docstring. * Fix grammar + clarify doc * Fix typo --------- Co-authored-by: Junpeng Lao <junpenglao@gmail.com>
Configuration menu - View commit details
-
Copy full SHA for f8db9aa - Browse repository at this point
Copy the full SHA f8db9aaView commit details
Commits on Jul 31, 2024
-
The jitted step remained unused, leading to the example running with an uncompiled nuts.step. Changing this reduces the execution time by a factor of 30 on my system and showcases blackjax' speed.
Configuration menu - View commit details
-
Copy full SHA for 441412a - Browse repository at this point
Copy the full SHA 441412aView commit details
Commits on Aug 7, 2024
-
Enable progress bar under pmap (#712)
* enable pmap progbar * fix bar creation * add locking * fix formatting * switch to using chain state
Configuration menu - View commit details
-
Copy full SHA for 27dfc9e - Browse repository at this point
Copy the full SHA 27dfc9eView commit details
Commits on Aug 9, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 148c028 - Browse repository at this point
Copy the full SHA 148c028View commit details
Commits on Aug 12, 2024
-
Simplify
run_inference_algorithm
(#714)* fix minor type errors * storing only expectation values * fixed memory efficient sampling * clean up * renaming vars * precommit fixes * fixing tests * fixing tests * fixing tests * fixing tests * fixing tests * merge main * burn in and fix tests * burn in and fix tests * minor fixes * minor fixes * minor fixes --------- Co-authored-by: jakob.robnik@gmail.com <jakob.robnik@gmail.com>
Configuration menu - View commit details
-
Copy full SHA for 7135fd7 - Browse repository at this point
Copy the full SHA 7135fd7View commit details
Commits on Aug 13, 2024
-
Configuration menu - View commit details
-
Copy full SHA for 834f55d - Browse repository at this point
Copy the full SHA 834f55dView commit details -
Configuration menu - View commit details
-
Copy full SHA for 4a11236 - Browse repository at this point
Copy the full SHA 4a11236View commit details