forked from pyro-ppl/numpyro
-
Notifications
You must be signed in to change notification settings - Fork 1
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
mock PR #2
Closed
Closed
mock PR #2
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…yro-ppl#1686) * add initial bug reproducer * disable arg validation during `tree_unflatten`
* Use Cholesky decomp instead of inverting kernel * Add a `use_cholesky` option * Use dash in args for consistency
…yro-ppl#1698) * make possible to fix the number of LF steps per sample and tune the step size * fix typo * fix version jaxns==2.2.6 * warn only when called from HMC * add test * lint
* scan replay handling * fix off by one error * cleaned a bit * lint * use substack * handle custom guides * handle empty value shape * a * test * move copy inside get_ith_value --------- Co-authored-by: frans <frans.zdyb@deondigital.com> Co-authored-by: OlaRonning <ronning@pm.me>
* add ruff * fix line-length * min python version * add requires-python section
* optional pre-commit hooks * add args
* initial config * fix sort * missing fix
* bound version * strong pin * add pin in setup.py * missing comma * fix bound * fix bound * add comma * more bounds * more bounds * more bounds * more bounds * rm unnecessary bound * rm pin bounds and unpin sphinx
* improve current ruff rules * rm comments * fix lengths * trim
* fix links in documentations * use latest sphinx search extension for security
* ensemble sampling draft * rewrite for loop as fori_loop * added efficiency comment for ESS GaussianMove * fix typo * fixed ravel for mixed dtype * add defaults * add support for potential_fn * AIES tests, warnings for AIES * AIES input validation * better docs, more input validation * ESS passing test cases * add tests for other files * linting * refactor ensemble_util * make test result less close to margin in CI, swap deprecated function * rename get_nondiagonal_indices, fix batch_ravel_pytree * print ensemble kernel diagnostics, smoke test parallel arg * fix docstring build * documentation * skip slow CI tests, unnest test if statements * fix doctest * doc rewrite * fix distribution test
* allow forward pass * fix params * add missing param in docstring * add flag to svi * typo docs * reorder arguments * order args * kw argument internal function * add arg to minimize * decouple aux function * rm kw argument unused * nicer doctrings * simple test * add wrapper * fix wrapper order * add wrapper trick to hmc
* Add `ReshapeTransform`. * Add `AutoBatchedMultivariateNormal`. * Refactor to use `AutoBatchedMixin`. * Add `AutoLowRankMultivariateNormal`. * Fix import order. * Disable batching along event dimensions.
* Initial bare bones implementation of DCC * Add tests and documentation * Make scale in Normal proposal configurable * Run linter * Add __init__.py file and allow parallel inference in tests * Move DCC tests to 'test chains' group
… and parallel chains (pyro-ppl#1746) * Remove jax tracers from MixedHMC.__getstate__() * Remove jax tracers from HMCGibbs serialization
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.