Skip to content
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
wants to merge 24 commits into from
Closed

mock PR #2

wants to merge 24 commits into from

Conversation

OlaRonning
Copy link
Member

No description provided.

pierreglaser and others added 24 commits November 24, 2023 22:35
…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
@OlaRonning OlaRonning closed this Feb 27, 2024
@OlaRonning OlaRonning deleted the main branch February 28, 2024 10:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.