.gitignore
.pre-commit-config.yaml
CHANGELOG.rst
README.rst
pyproject.toml
.github/workflows/ci.yml
.github/workflows/docs.yml
.github/workflows/pypi.yml
docs/.gitkeep
docs/_config.yml
docs/_toc.yml
docs/example.ipynb
docs/example.py
docs/model_out/loss.png
docs/model_out/reconstructions.png
docs/model_out/training_data.png
docs/out_mcmc/ci_plot.png
docs/out_mcmc/corner_plot.png
docs/out_mcmc/trace_plot.png
docs/studies/cyclical_beta/cyclical_beta.py
docs/studies/cyclical_beta/cyclical_beta.rst
docs/studies/cyclical_beta/out_models/beta_0/ci_plot.png
docs/studies/cyclical_beta/out_models/beta_0/loss.png
docs/studies/cyclical_beta/out_models/beta_0/training_reconstructions.gif
docs/studies/cyclical_beta/out_models/beta_1/ci_plot.png
docs/studies/cyclical_beta/out_models/beta_1/loss.png
docs/studies/cyclical_beta/out_models/beta_1/training_reconstructions.gif
docs/studies/cyclical_beta/out_models/beta_cyclical/ci_plot.png
docs/studies/cyclical_beta/out_models/beta_cyclical/loss.png
docs/studies/cyclical_beta/out_models/beta_cyclical/training_reconstructions.gif
docs/studies/cyclical_beta/out_models/beta_monotonic/ci_plot.png
docs/studies/cyclical_beta/out_models/beta_monotonic/loss.png
docs/studies/cyclical_beta/out_models/beta_monotonic/training_reconstructions.gif
docs/studies/evidence/compute_lnz.py
docs/studies/loss_vs_zsize/loss_vs_zsize.png
docs/studies/loss_vs_zsize/loss_vs_zsize.py
docs/studies/loss_vs_zsize/loss_vs_zsize.rst
src/starccato_jax/__init__.py
src/starccato_jax/_version.py
src/starccato_jax/config.py
src/starccato_jax/data.py
src/starccato_jax/io.py
src/starccato_jax/loss.py
src/starccato_jax/model.py
src/starccato_jax/trainer.py
src/starccato_jax.egg-info/PKG-INFO
src/starccato_jax.egg-info/SOURCES.txt
src/starccato_jax.egg-info/dependency_links.txt
src/starccato_jax.egg-info/requires.txt
src/starccato_jax.egg-info/top_level.txt
src/starccato_jax/plotting/__init__.py
src/starccato_jax/plotting/bayesian_plots.py
src/starccato_jax/plotting/gif_generator.py
src/starccato_jax/plotting/plotting.py
src/starccato_jax/sampler/__init__.py
src/starccato_jax/sampler/core.py
src/starccato_jax/sampler/gss_evidence.py
src/starccato_jax/sampler/sampler.py
src/starccato_jax/sampler/stepping_stone_evidence.py
src/starccato_jax/sampler/utils.py
tests/.gitkeep
tests/conftest.py
tests/test_evidence.py
tests/test_gif.py
tests/test_utils.py
tests/test_vae.py