LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
src/jaxns/__init__.py
src/jaxns/plotting.py
src/jaxns/public.py
src/jaxns/utils.py
src/jaxns/warnings.py
src/jaxns.egg-info/PKG-INFO
src/jaxns.egg-info/SOURCES.txt
src/jaxns.egg-info/dependency_links.txt
src/jaxns.egg-info/requires.txt
src/jaxns.egg-info/top_level.txt
src/jaxns/experimental/__init__.py
src/jaxns/experimental/evidence_maximisation.py
src/jaxns/experimental/global_optimisation.py
src/jaxns/experimental/public.py
src/jaxns/experimental/solvers/__init__.py
src/jaxns/experimental/solvers/ad_utils.py
src/jaxns/experimental/solvers/cg.py
src/jaxns/experimental/solvers/gauss_newton_cg.py
src/jaxns/experimental/solvers/test_cg.py
src/jaxns/experimental/solvers/tests/__init__.py
src/jaxns/experimental/solvers/tests/test_ad_utils.py
src/jaxns/experimental/solvers/tests/test_gauss_newton_cg.py
src/jaxns/experimental/tests/__init__.py
src/jaxns/experimental/tests/performance_tests.py
src/jaxns/experimental/tests/test_evidence_maximisation.py
src/jaxns/experimental/tests/test_global_optimisation.py
src/jaxns/framework/__init__.py
src/jaxns/framework/abc.py
src/jaxns/framework/bases.py
src/jaxns/framework/context.py
src/jaxns/framework/jaxify.py
src/jaxns/framework/model.py
src/jaxns/framework/ops.py
src/jaxns/framework/prior.py
src/jaxns/framework/special_priors.py
src/jaxns/framework/wrapped_tfp_distribution.py
src/jaxns/framework/tests/__init__.py
src/jaxns/framework/tests/test_context.py
src/jaxns/framework/tests/test_jaxify.py
src/jaxns/framework/tests/test_model.py
src/jaxns/framework/tests/test_prior.py
src/jaxns/internals/__init__.py
src/jaxns/internals/constraint_bijections.py
src/jaxns/internals/cumulative_ops.py
src/jaxns/internals/interp_utils.py
src/jaxns/internals/linalg.py
src/jaxns/internals/log_semiring.py
src/jaxns/internals/logging.py
src/jaxns/internals/maps.py
src/jaxns/internals/mixed_precision.py
src/jaxns/internals/namedtuple_utils.py
src/jaxns/internals/prefix_sum.py
src/jaxns/internals/pytree_utils.py
src/jaxns/internals/random.py
src/jaxns/internals/shapes.py
src/jaxns/internals/shrinkage_statistics.py
src/jaxns/internals/stats.py
src/jaxns/internals/tree_structure.py
src/jaxns/internals/types.py
src/jaxns/internals/tests/__init__.py
src/jaxns/internals/tests/test_constraint_bijections.py
src/jaxns/internals/tests/test_cumulative_ops.py
src/jaxns/internals/tests/test_interp_utils.py
src/jaxns/internals/tests/test_linalg.py
src/jaxns/internals/tests/test_log_semiring.py
src/jaxns/internals/tests/test_maps.py
src/jaxns/internals/tests/test_namedtuple_utils.py
src/jaxns/internals/tests/test_prefix_sum.py
src/jaxns/internals/tests/test_random.py
src/jaxns/internals/tests/test_shapes.py
src/jaxns/internals/tests/test_shrink_statistics.py
src/jaxns/internals/tests/test_stats.py
src/jaxns/internals/tests/test_tree_structure.py
src/jaxns/internals/tests/test_types.py
src/jaxns/nested_samplers/__init__.py
src/jaxns/nested_samplers/abc.py
src/jaxns/nested_samplers/common/__init__.py
src/jaxns/nested_samplers/common/initialisation.py
src/jaxns/nested_samplers/common/termination.py
src/jaxns/nested_samplers/common/types.py
src/jaxns/nested_samplers/common/uniform_sample.py
src/jaxns/nested_samplers/sharded/__init__.py
src/jaxns/nested_samplers/sharded/sharded_static.py
src/jaxns/nested_samplers/sharded/tests/__init__.py
src/jaxns/nested_samplers/sharded/tests/test_sharded_static.py
src/jaxns/samplers/__init__.py
src/jaxns/samplers/abc.py
src/jaxns/samplers/bases.py
src/jaxns/samplers/multi_ellipsoidal_samplers.py
src/jaxns/samplers/multi_slice_sampler.py
src/jaxns/samplers/uni_slice_sampler.py
src/jaxns/samplers/uniform_samplers.py
src/jaxns/samplers/multi_ellipsoid/__init__.py
src/jaxns/samplers/multi_ellipsoid/em_gmm.py
src/jaxns/samplers/multi_ellipsoid/multi_ellipsoid_utils.py
src/jaxns/samplers/multi_ellipsoid/tests/__init__.py
src/jaxns/samplers/multi_ellipsoid/tests/test_em_gmm.py
src/jaxns/samplers/multi_ellipsoid/tests/test_multi_ellipsoid.py
src/jaxns/tests/__init__.py
src/jaxns/tests/conftest.py
src/jaxns/tests/test_nested_sampler.py
src/jaxns/tests/test_utils.py