LICENSE
README.md
pyproject.toml
src/jaxns/__init__.py
src/jaxns/checkpoint.py
src/jaxns/cli.py
src/jaxns/constrained_sampler.py
src/jaxns/core.py
src/jaxns/cumulative_ops.py
src/jaxns/depth_condition.py
src/jaxns/distributed_core.py
src/jaxns/jaxify.py
src/jaxns/log_semiring.py
src/jaxns/logging.py
src/jaxns/mixed_precision.py
src/jaxns/model.py
src/jaxns/optional.py
src/jaxns/posterior.py
src/jaxns/priors.py
src/jaxns/pytree.py
src/jaxns/random_utils.py
src/jaxns/results.py
src/jaxns/run_config.py
src/jaxns/run_control.py
src/jaxns/samples.py
src/jaxns/state.py
src/jaxns/stats_utils.py
src/jaxns/types.py
src/jaxns.egg-info/PKG-INFO
src/jaxns.egg-info/SOURCES.txt
src/jaxns.egg-info/dependency_links.txt
src/jaxns.egg-info/entry_points.txt
src/jaxns.egg-info/requires.txt
src/jaxns.egg-info/top_level.txt
src/jaxns/algorithm/__init__.py
src/jaxns/algorithm/allocation.py
src/jaxns/algorithm/depth.py
src/jaxns/algorithm/initialisation.py
src/jaxns/algorithm/race_tree.py
src/jaxns/algorithm/schedule_storage.py
src/jaxns/algorithm/scheduler.py
src/jaxns/diagnostics/__init__.py
src/jaxns/diagnostics/plotting.py
src/jaxns/diagnostics/reference.py
src/jaxns/diagnostics/summary.py
src/jaxns/runtime/__init__.py
src/jaxns/runtime/client.py
src/jaxns/runtime/config.py
src/jaxns/runtime/coordinator.py
src/jaxns/runtime/node.py
src/jaxns/runtime/protocol.py
src/jaxns/runtime/session.py
src/jaxns/runtime/worker.py
src/jaxns/sampling/__init__.py
src/jaxns/sampling/batching.py
src/jaxns/sampling/continuation.py
src/jaxns/sampling/ellipsoid.py
src/jaxns/sampling/gmm.py
src/jaxns/sampling/prior.py
src/jaxns/sampling/protocol.py
src/jaxns/sampling/slice.py
src/jaxns/shrinkage/__init__.py
src/jaxns/shrinkage/classic.py
src/jaxns/shrinkage/phantom.py
src/jaxns/shrinkage/reference.py