├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md └── workflows │ └── unittests.yml ├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── benchmarks ├── difficult_problems │ └── main.py ├── gh117 │ ├── main.py │ └── run_benchmark.sh ├── gh123 │ ├── H_error_logZ_error_v2.6.2.png │ ├── main.py │ ├── results │ └── run_benchmark.sh ├── gh136 │ ├── Gradient_guided_vs_baseline_D16_v2.6.5.png │ ├── Gradient_guided_vs_baseline_D8_v2.6.5.png │ └── main.py ├── gh168 │ ├── main.py │ ├── results │ └── run_benchmark.sh └── parallel_problems │ ├── main.py │ └── results ├── debug ├── evidence_maximisation.ipynb └── scan_associative_weirdness.py ├── docs ├── Makefile ├── _templates │ └── autoapi │ │ └── python │ │ └── module.rst ├── conf.py ├── examples │ ├── Jones_scalar_modelling.ipynb │ ├── Lennard_Jones_potential.ipynb │ ├── constant_likelihood.ipynb │ ├── debug │ │ ├── dark_matter_model.ipynb │ │ ├── gaussian_process_marginalisation_spectral.ipynb │ │ ├── placebo_effect.ipynb │ │ └── plausible_logic.ipynb │ ├── dual_moon.ipynb │ ├── efficient_parameter_estimation.ipynb │ ├── egg_box.ipynb │ ├── gamma_poission.ipynb │ ├── gaussian_process_marginalisation.ipynb │ ├── gaussian_shells.ipynb │ ├── global_optimisation_neural_network.ipynb │ ├── gradient_guided.ipynb │ ├── intro_cornerplot.png │ ├── intro_diagnostics.png │ ├── intro_example.ipynb │ ├── mvn_data_mvn_prior.ipynb │ ├── nbconvert_all.sh │ ├── ou_process.ipynb │ └── self_exciting_point_process.ipynb ├── index.rst ├── make.bat ├── papers │ └── phantom-powered-nested-sampling │ │ ├── ablation_results_8D.npz │ │ ├── experiment_results_16D.npz │ │ ├── experiment_results_8D.npz │ │ ├── figures │ │ ├── ablation_bias_vs_k_8D.pdf │ │ ├── bias_vs_s_grid_per_c_16D.pdf │ │ ├── bias_vs_s_grid_per_c_8D.pdf │ │ ├── clique-analysis.drawio.png │ │ ├── sample_tree_linear.pdf │ │ ├── sample_tree_linear.pdf_tex │ │ ├── sample_tree_linear.svg │ │ ├── sample_tree_multi.pdf │ │ ├── sample_tree_multi.pdf_tex │ │ ├── sample_tree_multi.svg │ │ ├── speedup_and_likelihood_vs_f_per_k_16D.pdf │ │ └── speedup_and_likelihood_vs_f_per_k_8D.pdf │ │ ├── phantom_bias_tradeoff.ipynb │ │ ├── run_ablation.py │ │ └── run_experiment.py ├── requirements.txt └── user-guide │ └── installation.rst ├── jaxns_logo.png ├── pyproject.toml ├── pytest.ini ├── readthedocs.yaml ├── requirements-examples.txt ├── requirements-experimental.txt ├── requirements-tests.txt ├── requirements.txt ├── setup.py ├── src └── jaxns │ ├── __init__.py │ ├── experimental │ ├── __init__.py │ ├── evidence_maximisation.py │ ├── global_optimisation.py │ ├── public.py │ ├── solvers │ │ ├── __init__.py │ │ ├── ad_utils.py │ │ ├── cg.py │ │ ├── gauss_newton_cg.py │ │ ├── test_cg.py │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── test_ad_utils.py │ │ │ └── test_gauss_newton_cg.py │ └── tests │ │ ├── __init__.py │ │ ├── performance_tests.py │ │ ├── test_evidence_maximisation.py │ │ └── test_global_optimisation.py │ ├── framework │ ├── __init__.py │ ├── abc.py │ ├── bases.py │ ├── context.py │ ├── jaxify.py │ ├── model.py │ ├── ops.py │ ├── prior.py │ ├── special_priors.py │ ├── tests │ │ ├── __init__.py │ │ ├── test_context.py │ │ ├── test_jaxify.py │ │ ├── test_model.py │ │ └── test_prior.py │ └── wrapped_tfp_distribution.py │ ├── internals │ ├── __init__.py │ ├── constraint_bijections.py │ ├── cumulative_ops.py │ ├── interp_utils.py │ ├── linalg.py │ ├── log_semiring.py │ ├── logging.py │ ├── maps.py │ ├── mixed_precision.py │ ├── namedtuple_utils.py │ ├── prefix_sum.py │ ├── pytree_utils.py │ ├── random.py │ ├── shapes.py │ ├── shrinkage_statistics.py │ ├── stats.py │ ├── tests │ │ ├── __init__.py │ │ ├── test_constraint_bijections.py │ │ ├── test_cumulative_ops.py │ │ ├── test_interp_utils.py │ │ ├── test_linalg.py │ │ ├── test_log_semiring.py │ │ ├── test_maps.py │ │ ├── test_namedtuple_utils.py │ │ ├── test_prefix_sum.py │ │ ├── test_random.py │ │ ├── test_shapes.py │ │ ├── test_shrink_statistics.py │ │ ├── test_stats.py │ │ ├── test_tree_structure.py │ │ └── test_types.py │ ├── tree_structure.py │ └── types.py │ ├── nested_samplers │ ├── __init__.py │ ├── abc.py │ ├── common │ │ ├── __init__.py │ │ ├── initialisation.py │ │ ├── termination.py │ │ ├── types.py │ │ └── uniform_sample.py │ └── sharded │ │ ├── __init__.py │ │ ├── sharded_static.py │ │ └── tests │ │ ├── __init__.py │ │ └── test_sharded_static.py │ ├── plotting.py │ ├── public.py │ ├── samplers │ ├── __init__.py │ ├── abc.py │ ├── bases.py │ ├── multi_ellipsoid │ │ ├── __init__.py │ │ ├── em_gmm.py │ │ ├── multi_ellipsoid_utils.py │ │ └── tests │ │ │ ├── __init__.py │ │ │ ├── test_em_gmm.py │ │ │ └── test_multi_ellipsoid.py │ ├── multi_ellipsoidal_samplers.py │ ├── multi_slice_sampler.py │ ├── uni_slice_sampler.py │ └── uniform_samplers.py │ ├── tests │ ├── __init__.py │ ├── conftest.py │ ├── test_nested_sampler.py │ └── test_utils.py │ ├── utils.py │ └── warnings.py └── update_pypi.sh /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/workflows/unittests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/.github/workflows/unittests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/difficult_problems/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/difficult_problems/main.py -------------------------------------------------------------------------------- /benchmarks/gh117/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh117/main.py -------------------------------------------------------------------------------- /benchmarks/gh117/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh117/run_benchmark.sh -------------------------------------------------------------------------------- /benchmarks/gh123/H_error_logZ_error_v2.6.2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh123/H_error_logZ_error_v2.6.2.png -------------------------------------------------------------------------------- /benchmarks/gh123/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh123/main.py -------------------------------------------------------------------------------- /benchmarks/gh123/results: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh123/results -------------------------------------------------------------------------------- /benchmarks/gh123/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh123/run_benchmark.sh -------------------------------------------------------------------------------- /benchmarks/gh136/Gradient_guided_vs_baseline_D16_v2.6.5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh136/Gradient_guided_vs_baseline_D16_v2.6.5.png -------------------------------------------------------------------------------- /benchmarks/gh136/Gradient_guided_vs_baseline_D8_v2.6.5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh136/Gradient_guided_vs_baseline_D8_v2.6.5.png -------------------------------------------------------------------------------- /benchmarks/gh136/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh136/main.py -------------------------------------------------------------------------------- /benchmarks/gh168/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh168/main.py -------------------------------------------------------------------------------- /benchmarks/gh168/results: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh168/results -------------------------------------------------------------------------------- /benchmarks/gh168/run_benchmark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/gh168/run_benchmark.sh -------------------------------------------------------------------------------- /benchmarks/parallel_problems/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/parallel_problems/main.py -------------------------------------------------------------------------------- /benchmarks/parallel_problems/results: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/benchmarks/parallel_problems/results -------------------------------------------------------------------------------- /debug/evidence_maximisation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/debug/evidence_maximisation.ipynb -------------------------------------------------------------------------------- /debug/scan_associative_weirdness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/debug/scan_associative_weirdness.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_templates/autoapi/python/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/_templates/autoapi/python/module.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/examples/Jones_scalar_modelling.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/Jones_scalar_modelling.ipynb -------------------------------------------------------------------------------- /docs/examples/Lennard_Jones_potential.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/Lennard_Jones_potential.ipynb -------------------------------------------------------------------------------- /docs/examples/constant_likelihood.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/constant_likelihood.ipynb -------------------------------------------------------------------------------- /docs/examples/debug/dark_matter_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/debug/dark_matter_model.ipynb -------------------------------------------------------------------------------- /docs/examples/debug/gaussian_process_marginalisation_spectral.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/debug/gaussian_process_marginalisation_spectral.ipynb -------------------------------------------------------------------------------- /docs/examples/debug/placebo_effect.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/debug/placebo_effect.ipynb -------------------------------------------------------------------------------- /docs/examples/debug/plausible_logic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/debug/plausible_logic.ipynb -------------------------------------------------------------------------------- /docs/examples/dual_moon.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/dual_moon.ipynb -------------------------------------------------------------------------------- /docs/examples/efficient_parameter_estimation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/efficient_parameter_estimation.ipynb -------------------------------------------------------------------------------- /docs/examples/egg_box.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/egg_box.ipynb -------------------------------------------------------------------------------- /docs/examples/gamma_poission.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/gamma_poission.ipynb -------------------------------------------------------------------------------- /docs/examples/gaussian_process_marginalisation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/gaussian_process_marginalisation.ipynb -------------------------------------------------------------------------------- /docs/examples/gaussian_shells.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/gaussian_shells.ipynb -------------------------------------------------------------------------------- /docs/examples/global_optimisation_neural_network.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/global_optimisation_neural_network.ipynb -------------------------------------------------------------------------------- /docs/examples/gradient_guided.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/gradient_guided.ipynb -------------------------------------------------------------------------------- /docs/examples/intro_cornerplot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/intro_cornerplot.png -------------------------------------------------------------------------------- /docs/examples/intro_diagnostics.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/intro_diagnostics.png -------------------------------------------------------------------------------- /docs/examples/intro_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/intro_example.ipynb -------------------------------------------------------------------------------- /docs/examples/mvn_data_mvn_prior.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/mvn_data_mvn_prior.ipynb -------------------------------------------------------------------------------- /docs/examples/nbconvert_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/nbconvert_all.sh -------------------------------------------------------------------------------- /docs/examples/ou_process.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/ou_process.ipynb -------------------------------------------------------------------------------- /docs/examples/self_exciting_point_process.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/examples/self_exciting_point_process.ipynb -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/ablation_results_8D.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/ablation_results_8D.npz -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/experiment_results_16D.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/experiment_results_16D.npz -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/experiment_results_8D.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/experiment_results_8D.npz -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/ablation_bias_vs_k_8D.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/ablation_bias_vs_k_8D.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/bias_vs_s_grid_per_c_16D.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/bias_vs_s_grid_per_c_16D.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/bias_vs_s_grid_per_c_8D.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/bias_vs_s_grid_per_c_8D.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/clique-analysis.drawio.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/clique-analysis.drawio.png -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.pdf_tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.pdf_tex -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_linear.svg -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.pdf_tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.pdf_tex -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/sample_tree_multi.svg -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/speedup_and_likelihood_vs_f_per_k_16D.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/speedup_and_likelihood_vs_f_per_k_16D.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/figures/speedup_and_likelihood_vs_f_per_k_8D.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/figures/speedup_and_likelihood_vs_f_per_k_8D.pdf -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/phantom_bias_tradeoff.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/phantom_bias_tradeoff.ipynb -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/run_ablation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/run_ablation.py -------------------------------------------------------------------------------- /docs/papers/phantom-powered-nested-sampling/run_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/papers/phantom-powered-nested-sampling/run_experiment.py -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/user-guide/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/docs/user-guide/installation.rst -------------------------------------------------------------------------------- /jaxns_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/jaxns_logo.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/pyproject.toml -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/pytest.ini -------------------------------------------------------------------------------- /readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/readthedocs.yaml -------------------------------------------------------------------------------- /requirements-examples.txt: -------------------------------------------------------------------------------- 1 | scikit-learn 2 | optax -------------------------------------------------------------------------------- /requirements-experimental.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements-tests.txt: -------------------------------------------------------------------------------- 1 | scikit-learn 2 | networkx 3 | psutil 4 | pytest 5 | flake8 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/setup.py -------------------------------------------------------------------------------- /src/jaxns/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/__init__.py -------------------------------------------------------------------------------- /src/jaxns/experimental/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/__init__.py -------------------------------------------------------------------------------- /src/jaxns/experimental/evidence_maximisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/evidence_maximisation.py -------------------------------------------------------------------------------- /src/jaxns/experimental/global_optimisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/global_optimisation.py -------------------------------------------------------------------------------- /src/jaxns/experimental/public.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/public.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/ad_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/ad_utils.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/cg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/cg.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/gauss_newton_cg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/gauss_newton_cg.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/test_cg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/test_cg.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/tests/__init__.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/tests/test_ad_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/tests/test_ad_utils.py -------------------------------------------------------------------------------- /src/jaxns/experimental/solvers/tests/test_gauss_newton_cg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/solvers/tests/test_gauss_newton_cg.py -------------------------------------------------------------------------------- /src/jaxns/experimental/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/experimental/tests/performance_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/tests/performance_tests.py -------------------------------------------------------------------------------- /src/jaxns/experimental/tests/test_evidence_maximisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/tests/test_evidence_maximisation.py -------------------------------------------------------------------------------- /src/jaxns/experimental/tests/test_global_optimisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/experimental/tests/test_global_optimisation.py -------------------------------------------------------------------------------- /src/jaxns/framework/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/__init__.py -------------------------------------------------------------------------------- /src/jaxns/framework/abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/abc.py -------------------------------------------------------------------------------- /src/jaxns/framework/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/bases.py -------------------------------------------------------------------------------- /src/jaxns/framework/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/context.py -------------------------------------------------------------------------------- /src/jaxns/framework/jaxify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/jaxify.py -------------------------------------------------------------------------------- /src/jaxns/framework/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/model.py -------------------------------------------------------------------------------- /src/jaxns/framework/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/ops.py -------------------------------------------------------------------------------- /src/jaxns/framework/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/prior.py -------------------------------------------------------------------------------- /src/jaxns/framework/special_priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/special_priors.py -------------------------------------------------------------------------------- /src/jaxns/framework/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/framework/tests/test_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/tests/test_context.py -------------------------------------------------------------------------------- /src/jaxns/framework/tests/test_jaxify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/tests/test_jaxify.py -------------------------------------------------------------------------------- /src/jaxns/framework/tests/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/tests/test_model.py -------------------------------------------------------------------------------- /src/jaxns/framework/tests/test_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/tests/test_prior.py -------------------------------------------------------------------------------- /src/jaxns/framework/wrapped_tfp_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/framework/wrapped_tfp_distribution.py -------------------------------------------------------------------------------- /src/jaxns/internals/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/internals/constraint_bijections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/constraint_bijections.py -------------------------------------------------------------------------------- /src/jaxns/internals/cumulative_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/cumulative_ops.py -------------------------------------------------------------------------------- /src/jaxns/internals/interp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/interp_utils.py -------------------------------------------------------------------------------- /src/jaxns/internals/linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/linalg.py -------------------------------------------------------------------------------- /src/jaxns/internals/log_semiring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/log_semiring.py -------------------------------------------------------------------------------- /src/jaxns/internals/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/logging.py -------------------------------------------------------------------------------- /src/jaxns/internals/maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/maps.py -------------------------------------------------------------------------------- /src/jaxns/internals/mixed_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/mixed_precision.py -------------------------------------------------------------------------------- /src/jaxns/internals/namedtuple_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/namedtuple_utils.py -------------------------------------------------------------------------------- /src/jaxns/internals/prefix_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/prefix_sum.py -------------------------------------------------------------------------------- /src/jaxns/internals/pytree_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/pytree_utils.py -------------------------------------------------------------------------------- /src/jaxns/internals/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/random.py -------------------------------------------------------------------------------- /src/jaxns/internals/shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/shapes.py -------------------------------------------------------------------------------- /src/jaxns/internals/shrinkage_statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/shrinkage_statistics.py -------------------------------------------------------------------------------- /src/jaxns/internals/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/stats.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_constraint_bijections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_constraint_bijections.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_cumulative_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_cumulative_ops.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_interp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_interp_utils.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_linalg.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_log_semiring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_log_semiring.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_maps.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_namedtuple_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_namedtuple_utils.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_prefix_sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_prefix_sum.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_random.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_shapes.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_shrink_statistics.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_stats.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_tree_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_tree_structure.py -------------------------------------------------------------------------------- /src/jaxns/internals/tests/test_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tests/test_types.py -------------------------------------------------------------------------------- /src/jaxns/internals/tree_structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/tree_structure.py -------------------------------------------------------------------------------- /src/jaxns/internals/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/internals/types.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/__init__.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/abc.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/common/__init__.py: -------------------------------------------------------------------------------- 1 | from jaxns.nested_samplers.common.types import * -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/common/initialisation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/common/initialisation.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/common/termination.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/common/termination.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/common/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/common/types.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/common/uniform_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/common/uniform_sample.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/sharded/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/sharded/__init__.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/sharded/sharded_static.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/sharded/sharded_static.py -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/sharded/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/nested_samplers/sharded/tests/test_sharded_static.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/nested_samplers/sharded/tests/test_sharded_static.py -------------------------------------------------------------------------------- /src/jaxns/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/plotting.py -------------------------------------------------------------------------------- /src/jaxns/public.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/public.py -------------------------------------------------------------------------------- /src/jaxns/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/__init__.py -------------------------------------------------------------------------------- /src/jaxns/samplers/abc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/abc.py -------------------------------------------------------------------------------- /src/jaxns/samplers/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/bases.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/em_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_ellipsoid/em_gmm.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/multi_ellipsoid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_ellipsoid/multi_ellipsoid_utils.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/tests/test_em_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_ellipsoid/tests/test_em_gmm.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoid/tests/test_multi_ellipsoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_ellipsoid/tests/test_multi_ellipsoid.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_ellipsoidal_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_ellipsoidal_samplers.py -------------------------------------------------------------------------------- /src/jaxns/samplers/multi_slice_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/multi_slice_sampler.py -------------------------------------------------------------------------------- /src/jaxns/samplers/uni_slice_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/uni_slice_sampler.py -------------------------------------------------------------------------------- /src/jaxns/samplers/uniform_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/samplers/uniform_samplers.py -------------------------------------------------------------------------------- /src/jaxns/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/jaxns/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/tests/conftest.py -------------------------------------------------------------------------------- /src/jaxns/tests/test_nested_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/tests/test_nested_sampler.py -------------------------------------------------------------------------------- /src/jaxns/tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/tests/test_utils.py -------------------------------------------------------------------------------- /src/jaxns/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/utils.py -------------------------------------------------------------------------------- /src/jaxns/warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/src/jaxns/warnings.py -------------------------------------------------------------------------------- /update_pypi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Joshuaalbert/jaxns/HEAD/update_pypi.sh --------------------------------------------------------------------------------