.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/study_impact_of_z.py
docs/model_exploration/loss_vs_z.png
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
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/plotting.py
src/starccato_jax/sampler.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
tests/.gitkeep
tests/conftest.py
tests/test_vae.py