.gitignore
.pdbrc
.readthedocs.yml
LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
.github/workflows/ci.yml
.github/workflows/docs.yml
.github/workflows/publish-to-pypi.yml
diffuse/base_forward_model.py
diffuse/predictor.py
diffuse/denoisers/__init__.py
diffuse/denoisers/base.py
diffuse/denoisers/denoiser.py
diffuse/denoisers/pnp_denoiser.py
diffuse/denoisers/utils.py
diffuse/denoisers/cond/__init__.py
diffuse/denoisers/cond/base.py
diffuse/denoisers/cond/dps.py
diffuse/denoisers/cond/fps.py
diffuse/denoisers/cond/tmp.py
diffuse/diffusion/__init__.py
diffuse/diffusion/sde.py
diffuse/examples/flux_dev/README.md
diffuse/examples/flux_dev/__init__.py
diffuse/examples/flux_dev/run_flux_inference.py
diffuse/examples/flux_dev/models/__init__.py
diffuse/examples/flux_dev/models/clip_text.py
diffuse/examples/flux_dev/models/flux_blocks.py
diffuse/examples/flux_dev/models/flux_transformer.py
diffuse/examples/flux_dev/models/flux_vae.py
diffuse/examples/flux_dev/models/mlp.py
diffuse/examples/flux_dev/models/t5_encoder.py
diffuse/examples/flux_dev/models/utils.py
diffuse/examples/flux_dev/utils/__init__.py
diffuse/examples/flux_dev/utils/flux_checkpoint.py
diffuse/examples/flux_dev/utils/flux_timer.py
diffuse/examples/gaussian_mixtures/cond_mixture.py
diffuse/examples/gaussian_mixtures/initialization.py
diffuse/examples/gaussian_mixtures/mixture.py
diffuse/examples/gaussian_mixtures/mixture_evolution.py
diffuse/examples/gaussian_mixtures/plotting.py
diffuse/examples/gaussian_mixtures/forward_models/matrix_product.py
diffuse/examples/mnist/forward_model.py
diffuse/integrator/__init__.py
diffuse/integrator/base.py
diffuse/integrator/deterministic.py
diffuse/integrator/stochastic.py
diffuse/neural_network/__init__.py
diffuse/neural_network/blocks/__init__.py
diffuse/neural_network/blocks/attention.py
diffuse/neural_network/blocks/decoder.py
diffuse/neural_network/blocks/downsample.py
diffuse/neural_network/blocks/encoder.py
diffuse/neural_network/blocks/resnet_block.py
diffuse/neural_network/blocks/time_embedding.py
diffuse/neural_network/blocks/timestep.py
diffuse/neural_network/blocks/upsample.py
diffuse/neural_network/nn/__init__.py
diffuse/neural_network/nn/condUNet.py
diffuse/neural_network/nn/params.py
diffuse/neural_network/nn/sdVae.py
diffuse/timer/__init__.py
diffuse/timer/base.py
diffuse/utils/__init__.py
diffuse/utils/logger.py
diffuse/utils/mapping.py
diffuse_jax.egg-info/PKG-INFO
diffuse_jax.egg-info/SOURCES.txt
diffuse_jax.egg-info/dependency_links.txt
diffuse_jax.egg-info/requires.txt
diffuse_jax.egg-info/top_level.txt
docs/Makefile
docs/api.rst
docs/build.py
docs/conf.py
docs/diffusion_crash_course.rst
docs/diffusion_tutorial.ipynb
docs/flux_tutorial.ipynb
docs/index.rst
docs/mnist_tutorial.ipynb
docs/quickstart.rst
docs/references.bib
docs/_static/.gitkeep
docs/_static/favicon.ico
docs/_static/logo.png
docs/api/cond_denoisers.rst
docs/api/denoiser.rst
docs/api/denoisers.rst
docs/api/diffusion.rst
docs/api/forward_models.rst
docs/api/integrators.rst
docs/api/neural_networks.rst
docs/api/timer.rst
docs/api/utilities.rst
docs/assets/denoised_image.gif
test/__init__.py
test/config.py
test/conftest.py
test/test_diffusion.py
test/test_image_forward.py
test/test_mnist.py
test/test_utils.py