├── .github └── workflows │ └── tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── HISTORY.rst ├── LICENSE ├── README.md ├── mypy.ini ├── noxfile.py ├── pyproject.toml ├── setup.py ├── src └── emcee_jax │ ├── __init__.py │ ├── _src │ ├── __init__.py │ ├── ensemble.py │ ├── host_callback.py │ ├── log_prob_fn.py │ ├── moves │ │ ├── __init__.py │ │ ├── core.py │ │ ├── slice.py │ │ └── util.py │ ├── ravel_util.py │ ├── sampler.py │ └── types.py │ ├── experimental │ ├── __init__.py │ └── moves │ │ ├── __init__.py │ │ └── hmc.py │ ├── host_callback.py │ └── moves.py └── tests ├── test_host_callback.py ├── test_moves.py ├── test_ravel_util.py └── test_sampler.py /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /HISTORY.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/HISTORY.rst -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/README.md -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/mypy.ini -------------------------------------------------------------------------------- /noxfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/noxfile.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/setup.py -------------------------------------------------------------------------------- /src/emcee_jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/__init__.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/emcee_jax/_src/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/ensemble.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/host_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/host_callback.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/log_prob_fn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/log_prob_fn.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/moves/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/moves/__init__.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/moves/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/moves/core.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/moves/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/moves/slice.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/moves/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/moves/util.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/ravel_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/ravel_util.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/sampler.py -------------------------------------------------------------------------------- /src/emcee_jax/_src/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/_src/types.py -------------------------------------------------------------------------------- /src/emcee_jax/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/emcee_jax/experimental/moves/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/emcee_jax/experimental/moves/hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/experimental/moves/hmc.py -------------------------------------------------------------------------------- /src/emcee_jax/host_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/host_callback.py -------------------------------------------------------------------------------- /src/emcee_jax/moves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/src/emcee_jax/moves.py -------------------------------------------------------------------------------- /tests/test_host_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/tests/test_host_callback.py -------------------------------------------------------------------------------- /tests/test_moves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/tests/test_moves.py -------------------------------------------------------------------------------- /tests/test_ravel_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/tests/test_ravel_util.py -------------------------------------------------------------------------------- /tests/test_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dfm/emcee-jax/HEAD/tests/test_sampler.py --------------------------------------------------------------------------------